我在深度学习中使用CIFAR-10数据集，但我想仅针对水果类别指定我的数据集。我们知道我们使用的是：

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

来加载所有的CIFAR-10数据集。如何仅加载水果类别的数据，而不是所有数据呢？

回答：

如果你不介意加载额外的数据，最简单的方法是找出水果的标签，然后做类似这样的事情：X_train, y_train = X_train[y_train == fruit_label], y_train[y_train == fruit_label]，前提是你的数据存储在np.arrays中。对于测试集也是类似的操作。

如果不行，那么你将不得不修改你的hdf5文件或你存储数据的任何地方。