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()