# 导入包
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/",one_hot=True)
这个点进运行可能会有一个比较长的时间,所以,我先下载好需要的数据集
网址是:http://yann.lecun.com/exdb/mnist/
点击图中四个红色链接即可下载,下好后是这样的
找到 jupyter 的文件存放目录,这里我是自己建的一个jupyter_project 专门存放jupyter中的文件,在里面新建一个MNIST_data文件夹,将四个压缩包直接粘贴进去。关于修改jupyter工作目录可以看我的这篇文章https://blog.csdn.net/weixin_43951831/article/details/105127406
再执行开头那段代码,导入数据集,速度会快一些
结果中出现Extracting xxx ,表示在解压缩读取里面的内容,所以基本上MNIST的数据集也就加载了
可以试一下查看里面的内容
print('训练集 train 数量:',mnist.train.num_examples,
',验证集 validation 数量:',mnist.validation.num_examples,
',测试集 test 数量:',mnist.test.num_examples)