下载并使用MNIST数据集
TensorFlow提供了一个库,可以直接用来自动下载与安装MNIST。
MNIST里包含3个数据集:第一个是训练数据集(mnist.train.images),另外两个分别是测试数据集(mnist.test.images)和验证数据集(mnist.validation)。
代码中的one_hot=True,表示将样本标签转化为one_hot编码。
刚开始的打印信息是解压数据集的意思。如果是第一次运行,还会显示下载数据的相关消息。
接着打印出来的是训练集的图片信息,是一个55000行、784列的矩阵。即,训练集里有55000张图片。
1 from tensorflow.examples.tutorials.mnist import input_data 2 mnist = input_data.read_data_sets("MNIST_data/",one_hot=True) 3 print ('输入数据:',mnist.train.images) 4 print ('输入打印shape:',mnist.train.images.shape) 5 import pylab 6 im = mnist.train.images[1] 7 im = im.reshape(-1,28) 8 pylab.imshow(im) 9 pylab.show()
代码的输出结果如图: