项目场景:
由于网上对TensorFlow2.3.1的相关操作介绍的很少,即使有大部分都是有错误的,于是我就通过自己的摸索,对此作以总结。
问题描述:
首先是通过tensorflow对MNIST数据集进行下载,网上对此具体的操作是使用以下方法进行下载:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data',one_hot = True)
原因分析:
因为TensorFlow2.0往后的版本已经更新了很多函数库,比如examples库中没有tutorials函数,如果非要使用可以对此进行下载。我觉得没必要,因为TensorFlow2.3.1中给我们提供了keras库,该库中给我们专门提供了对相关数据集的下载,具体的操作在下方的完整代码中。
解决方案:
首先是对MNIST数据集的可视化显示完整代码:
接下来是对MNIST数据集做的相关训练完整代码: