實現一個簡單的網絡訓練---tensorflow2.0學習

處理數據 dataset = tf.data.Dataset.from_tensor_slices((train_x, train_y)) dataset = dataset.shuffle(4)#打亂,越大越亂 dataset = dataset.batch(32)#batch——size dataset = dataset.repeat()#數據集重複次數,通常默認爲空,其實就是epoch #
相關文章
相關標籤/搜索