如何从文件加载权重并使用它们来预测Keras中的测试数据

时间:2018-12-02 12:06:23

标签: keras

昨天晚上,我接受了神经网络模型的训练,这很花时间,所以我想添加一条语句来减轻体重model.save_weights('first_try.h5')

现在,有了文件,我想从保存的文件中受益。 预测就像

pred=model.predict_generator(test_generator, steps=4124, verbose=1)

1 个答案:

答案 0 :(得分:1)

如果保存了模型的权重,则可以使用load_weights方法加载。但是首先,您必须定义模型结构。

例如

model = method_to_create_the_model()
model.load_weights("path_to_weight_file")