我正在做一个深度学习项目。训练后,我将模型另存为h5。在另一个文件中,我加载保存的模型并使用该模型进行预测。但是,当我在Pycharm中运行代码时,模型将再次开始训练。我重新启动笔记本电脑,然后再次运行,但仍然出现相同的现象。 pycharm是否在错误的文件上运行?
model.save('model_10000.h5')
然后在另一个文件中
model = load_model('model_10000.h5')
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'])
# predict for test set
pred = model.predict(testX)
答案 0 :(得分:0)
您不需要编译保存的模型,也许与此有关。
model.save('model_10000.h5')
model = load_model('model_10000.h5')
pred = model.predict(testX)
请检查此以获取更多详细信息:https://www.tensorflow.org/tutorials/keras/save_and_load
答案 1 :(得分:0)
您无需重新编译模型。您保存的模型始终会被编译,当您将其加载回时,它始终会返回编译后的模型(refer Keras FAQ)。
因此,只需删除model.compile
步骤,您就会很方便。