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为自定义预处理函数。

 循环训练:

 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM