带有Sklearn的MNIST数据集

Question

我正在MNIST数据集上训练线性模型，但我只想训练一个数字4。如何选择X_test，X_train，y_test，y_train？

Answer 1

您的分类器需要学习区分不同类别的集合。如果您只关心数字4，则应将训练和测试集划分为：

• 第4类实例
• 非第4类实例：所有其他数字的并集

否则，火车/测试拆分仍是典型的，您不希望重叠的部分。