PyTorch入门实战教程笔记(三):手写数字问题引入
MNIST数据集
MNIST数据集是机器学习领域中非常经典的一个数据集,由60000个训练样本和10000个测试样本组成,每个样本都是一张28 * 28像素的灰度手写数字图片。
下载数据集
网址:http://yann.lecun.com/exdb/mnist/
数据集简介:
1、共有4数据集,下载之后保存在磁盘中(最好放在代码执行目录下,方便后期使用。)如新建一个文件夹D:*****\MNIST_data存放数据。
2、数据集中,训练样本:共60000个,其中55000个用于训练,另外5000个用于验证,测试样本:共10000个
Input Or Output
输入:
前面介绍了MNIST数据集中每个样本为28*28个像素,用矩阵表示即[28,28],将其打平为一个维度(后一行接在前一行的后面),即[784],也就是此时x是一个一维度的矩阵。那么这样子,输入x为[b,784],这里的b代表有多少证张图片,例如有一张,b=1.
输出&