小编S.H*_*viv的帖子

我想使用 CIFAR-10 数据集，但我只想要青蛙、狗、猫、马和鸟类，到目前为止我已经使用了以下代码：

  # Plot ad hoc CIFAR10 instances
  from keras.datasets import cifar10
  from matplotlib import pyplot
  from scipy.misc import toimage
  # load data
  (X_train, y_train), (X_test, y_test) = cifar10.load_data()
  # create a grid of 3x3 images
  for i in range(0, 9):
      pyplot.subplot(330 + 1 + i)
      pyplot.imshow(toimage(X_train[i]))
  # show the plot
  pyplot.show()

cifar10.load_data() 函数加载整个数据，我可以只获取所需的类吗？