1.下载MNIST数据集,下载文件并保存到本地目录(即下载下面四个文件)
2.定义读取函数:我使用的是numpy来读取二进制文件
3.读取数据:调用读取函数加载训练集和测试集。(下面是部分代码)
def load_mnist(path, kind='train'):
"""Load MNIST data from `path`"""
if kind == 'train':
labels_path = os.path.join(path, 'train-labels.idx1-ubyte')
images_path = os.path.join(path, 'train-images.idx3-ubyte')
else:
labels_path = os.path.join(path, 't10k-labels.idx1-ubyte')
images_path = os.path.join(path, 't10k-images.idx3-ubyte')
4.显示数据:使用matplotlib显示一张手写数字图像。
这样输入全部代码就可以有结果图了。
希望对你有所帮助。