MNIST 数据集

加载MNIST数据

为了方便起见，我们已经准备了一个脚本来自动下载和导入MNIST数据集。它会自动创建一个 'MNIST_data' 的目录来存储数据。

import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)

这里，mnist是一个轻量级的类。它以Numpy数组的形式存储着训练、校验和测试数据集。同时提供了一个函数，用于在迭代中获得minibatch，后面我们将会用到。