原标题：下载并使用MNIST数据集

原文来自：博客园      原文链接：https://www.cnblogs.com/Jesse-Cavendish/p/11646985.html

TensorFlow提供了一个库，可以直接用来自动下载与安装MNIST。

MNIST里包含3个数据集：第一个是训练数据集（mnist.train.images），另外两个分别是测试数据集(mnist.test.images)和验证数据集（mnist.validation）。

代码中的one_hot=True，表示将样本标签转化为one_hot编码。
刚开始的打印信息是解压数据集的意思。如果是第一次运行，还会显示下载数据的相关消息。
接着打印出来的是训练集的图片信息，是一个55000行、784列的矩阵。即，训练集里有55000张图片。

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/",one_hot=True)
print ('输入数据:',mnist.train.images)
print ('输入打印shape:',mnist.train.images.shape)
import pylab
im = mnist.train.images[1]
im = im.reshape(-1,28)
pylab.imshow(im)
pylab.show()

代码的输出结果如图：

免责声明：本文来自互联网新闻客户端自媒体，不代表本网的观点和立场。

合作及投稿邮箱：E-mail:editor@tusaishared.com