python - 如何在 Keras 中保存经过训练的模型以便在应用程序中使用它？

Question

我在 Keras 中训练了一个模型，并以不同的方式保存它，例如；

model.save("filename")

或

model.to_json()  
model.save_weights("filename")

但是当我在另一个程序中加载经过训练的模型来进行预测时，我得到的结果与测试结果非常不同。

为什么会发生这种情况以及我该如何处理？

Answer 1

保存如下:

     model.save('model.h5')
     model_json = model.to_json()
     with open("model.json", "w") as json_file:
         json_file.write(model_json)

然后，为了有效地将其加载到应用程序中，请将其设置为如下所示的全局变量，以便它不会一次又一次加载:

    def load_model():

        global model

        json_file = open('model.json', 'r')
        model_json = json_file.read()
        model = model_from_json(model_json)
        model.load_weights("model.h5")
        model._make_predict_function()