TensorFlow实现mnist书写数字分类,
在使用
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('./data/mnist', one_hot=True)
- 1
- 2
导入mnist数据集时,无法下载。
出现please use urllib or similar directly错误。
具体内容如下:
解决方法
在 from tensorflow.examples.tutorials.mnist import input_data
之前,加入
old_v = tf.logging.get_verbosity()
tf.logging.set_verbosity(tf.logging.ERROR)
- 1
- 2
在mnist = input_data.read_data_sets('./data/mnist', one_hot=True)
之后,加入
tf.logging.set_verbosity(old_v)
- 1
再次运行,会经过大概5分钟左右的时间下载mnist数据集。
下载完成后,项目目录中会出现
参考链接:https://codeday.me/bug/20181231/475924.html
另外,还可以手动下载mnist数据集,放到项目目录下。
在这个链接下,对应的下载四个文件即可 。