MNIST是机器学习中“hello world”式的入门级别的计算机视觉数据集,它包含各种手写数字图片。它也包含每一张图片对应的标签,告诉我们这个数字是几。
我们将训练一个机器模型用于预测图片里面的数字。先从一个简单的数学模型开始:Softmax Regression。
对应这个教程的实现代码很短,而且真正有意思的内容只包含在三行代码中。但是去理解包含在这些代码里面的设计思想是非常重要的:Tensorflow工作流程和机器学习的基本概念。
MNIST数据集的下载和安装:
用下面的代码导入到你的项目里面,可以下载这份代码,也可以直接复制粘贴到你的代码文件里面。
import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
(在这之后就会下载相应的训练集和测试集的样本)