昨天晚上,我接受了神经网络模型的训练,这很花时间,所以我想添加一条语句来减轻体重model.save_weights('first_try.h5')
现在,有了文件,我想从保存的文件中受益。 预测就像
pred=model.predict_generator(test_generator, steps=4124, verbose=1)
答案 0 :(得分:1)
如果保存了模型的权重,则可以使用load_weights方法加载。但是首先,您必须定义模型结构。
例如
model = method_to_create_the_model()
model.load_weights("path_to_weight_file")