Tensorflow2.0学习(4)---数据集Dataset

一、经典数据集

keras.datasets:

加载:

datasets.XXX.load_data():实现经典书籍集的自动加载,XXX表示数据集名称,如MNIST。返回两个tuple,第一个tuple为训练集(x,y),第二个tuple为测试集(x,y),用Numpy数组保存。

tf.data.Dataset.from_tensor_slices:将训练部分数据图片x和标签y都转换成Dataset对象。将数据转换成 Dataset 对象后,一般需要再添加一系列的数据集标准处理步骤,如随机打散、预处理、按批装载等。

随机打散:

Dataset.shuffle(buffer_size):随机打散数据之间的顺序。

批训练:

Dataset.batch(batch_size):取batch_size大小的数据集。

预处理:

Dataset.map(func):func为自定义预处理函数。

 循环训练:

 

原文地址:https://www.cnblogs.com/Lee-yl/p/12566877.html