利用tensorflow2.0下载MNIST数据集
import tensorflow as tf
data = tf.keras.datasets.mnist
(train_x, train_y), (test_x, test_y) = data.load_data()
一般会出现如下报错:
原因是因为网络下载出错,解决方法是提前下载数据,MNIST下载链接为
https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
并将下载好的mnist.npz放置到
/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/keras/datasets目录下。我是3.7版本,版本差异自行修改目录。
接着修改报错信息中的mnist.py文件点击进入文件
将原本网络链接部分注释,修改path为下载好的mnist.npz的目录地址。
运行结果
import tensorflow as tf
data = tf.keras.datasets.mnist
(train_x, train_y), (test_x, test_y) = data.load_data()
print(train_x.shape)
windows下解决方案基本一致,注意目录地址即可。