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

问题描述 投票:0回答:1

我将keras DNN模型保存为'.h5'文件,你能告诉我,用于保存keras模型的其他方法是什么。根据性能让我知道这些文件。

谢谢

python-3.x keras model neural-network
1个回答
0
投票

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

JSON

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

Yamla

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

由于这些需要额外的格式化步骤,我希望你能从标准的model.save_weights获得最佳性能。

© www.soinside.com 2019 - 2024. All rights reserved.