如何根据特定的类名加载CIFAR-10数据集？

Question

我使用CIFAR-10数据集进行深入学习，但是我只想为水果类指定我的数据集。我们知道我们使用了：

(X_train, y_train), (X_test, y_test) = cifar10.load_data() 

加载所有CIFAR-10数据集。如何仅为水果类加载数据而不是所有数据？

Answer 1

如果您不介意加载其他数据，最简单的方法是找出女巫是水果标签，并执行如下操作：X_train, y_train = X_train[y_train == fruit_label], y_train[y_train == fruit_label]，前提是您的数据存储在np.arrays中。等同于您的测试集。

如果没有，那么你将不得不修改你的hdf5文件或无论你在哪里存储你的数据。