使用已儲存的權重對測試集進行預測
儲存在 hdf5 檔案中的模型權重可以重複利用,載入到尚未訓練的模型中。一旦將這些權重載入模型後,它的行為就和一個已訓練至相同權重的模型相同。例如,你可以用這個模型對未看過的資料集(例如 test_data)進行預測。
本練習屬於課程
使用 Keras 進行影像建模
練習說明
- 從名為
'weights.hdf5'的檔案載入權重。 - 預測
test_data中前 3 張影像的類別。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Load the weights from file
model.____
# Predict from the first three images in the test data
____