如何子集MNIST数据集？

Question

我正在关注TensorFlow的Generative Adversarial Network的教程。本教程使用MNIST数据集来训练模型。我想减少输入的大小，以便我的程序运行速度更快，但不知道如何获取我正在使用的MNIST数据集的子集。下面是我用于提取所述数据集的代码：

from tensorflow.examples.tutorials.mnist import input_data 
mnist = input_data.read_data_sets("MNIST_data/") 

Answer 1

有一种方法

mnist.next_batch(batchsize) 

提取从列车组的长度BATCHSIZE的随机样本。

如果你不想要的东西是随机的，您可以通过

x = mnist.train.images[start_batch:end_batch] 
y = mnist.train.labels[start_batch:end_batch] 

或类似与mnist.test访问它们的测试集。