下载并使用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()


代码的输出结果如图:

posted @ 2019-10-10 11:41  Weisswire  阅读(13268)  评论(2编辑  收藏  举报