pytorch学习(二):mnist手写数字数据集训练和测试(一)mnist数据集分析
MNIST 数据集来自美国国家标准与技术研究所, National Institute of Standards and Technology (NIST). 训练集 (training set) 由来自 250 个不同人手写的数字构成, 其中 50% 是高中学生, 50% 来自人口普查局 (the Census Bureau) 的工作人员. 测试集(test set) 也是同样比例的手写数字数据. MNIST 数据集可在 http://yann.lecun.com/exdb/mnist/ 获取,将下载来的数据解压保存到datesets文件下,四个文件分别包括了images和labels,其中labels是images所对应的数字,训练集为50000个样本,测试集为10000个样本。 像素为28x28,单通道 图片是以字节的形式进行存储,需要把它转换成矩阵形式, import os import struct import numpy as np def load_mnist(path, kind='train'): """Load MNta from `path`""" labels_path = os.path.join(path,IST da '%s-labels.idx1-ubyte' % kind) images_path = os.path.join(path, '%s