Keras加载模型总是训练而不是预测

时间:2020-04-17 03:46:13

标签: machine-learning keras

我正在做一个深度学习项目。训练后,我将模型另存为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)

这就是我得到的 enter image description here

2 个答案:

答案 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步骤,您就会很方便。