有不同的方法来保存keras DNN模型吗?

时间:2019-04-24 11:16:06

标签: python-3.x keras model neural-network

我正在将keras DNN模型保存为“ .h5”文件,您能告诉我,用于保存keras模型的所有其他方法是什么。根据性能,我知道那些文件。

谢谢

1 个答案:

答案 0 :(得分:0)

您可以保存/加载描述为h5的JSON或YAML。

JSON

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

YAML

model_yaml = model.to_yaml()
with open("model.yaml", "w") as yaml_file:
    yaml_file.write(model_yaml)

由于这些操作需要额外的格式化步骤,因此我希望您将从标准model.save_weights中获得最佳性能。