在Tensorflow 2 中使用input_data模块导入数据

在学习Tensorflow的过程中,参照的是Morvan的Tensorflow入门视频,但是他采用的是Tensorflow 1版本的代码。由于本人学术不精,就想将Tensorflow 2转换成Tensorflow 1。因此查阅大量资料,目前的方法主要是采用以下两行代码:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

替换了 import tensorflow as tf

但是就算使用了上面两行代码,依然无法使用input_data模块,参照Morvan的代码如下:

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
运行后出现如下问题:

ModuleNotFoundError: No module named 'tensorflow.examples'

也就是在Tensorflow 2中未包括tensorflow.examples模块。

因此采用降版本的方法:

 import tensorflow.compat.v1 as tf
from tensorflow.examples.tutorials.mnist import input_data
tf.disable_v2_behavior()
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
但是仍然存在以下错误:

ModuleNotFoundError: No module named 'tensorflow.examples'

说明 from tensorflow.examples.tutorials.mnist import input_data 这句代码无法使用在降版本后的Tensorflow中

我们发现只是需要input_data这个模块,因此,我们考虑在python项目文件夹中创建一个input_data.py文件,然后在Classification分类学习.py文件中导入input_data:

在Tensorflow 2 中使用input_data模块导入数据_第1张图片

代码如下:

import tensorflow.compat.v1 as tf
# from tensorflow.examples.tutorials.mnist import input_data
tf.disable_v2_behavior()
import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)

剩余代码参照Morvan的Tensorflow视频,运行结果:

在Tensorflow 2 中使用input_data模块导入数据_第2张图片

 关于input_data.py,CSDN中也有其他大佬总结了,可以直接搜索。

新手一枚,大佬勿喷!!!

 

你可能感兴趣的:(python)