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