如何对 MNIST 数据集进行子集化？

Question

我正在关注有关 TensorFlow 的生成对抗网络的教程。本教程使用 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 类似的测试集。