下载并使用MNIST数据集

匿名 (未验证) 提交于 2019-12-03 00:14:01

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()


代码的输出结果如图:

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!