使用已保存的权重在测试集上进行预测
存储在 hdf5 文件中的模型权重可以复用于一个未训练的模型。将这些权重加载到模型后,模型的行为与训练至该权重的模型相同。例如,您可以用该模型对未见过的数据集(如 test_data)进行预测。
本练习是课程的一部分
使用 Keras 进行图像建模
练习说明
- 从名为
'weights.hdf5'的文件加载权重。 - 预测
test_data中前三张图像的类别。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Load the weights from file
model.____
# Predict from the first three images in the test data
____