代码之家 › 专栏 › 技术社区 › Baron Yugovich

Keras to_json(),它能保存什么?

keras json python

Baron Yugovich · 技术社区 · 6 年前

我的印象是它只保存了模型的架构,所以我应该能够在开始训练之前调用它?然后 save_weights()

在什么阶段我可以叫 to_json() ? 一、我要打电话吗 compile() fit() ?

1 回复 | 直到 6 年前

today 6 年前

如中所述 Keras docs 它只保存模型的体系结构:

仅保存/加载模型的体系结构

如果只需要保存模型的体系结构,而不需要重量或其训练配置,您可以:
# save as JSON
json_string = model.to_json()

# save as YAML
yaml_string = model.to_yaml()
生成的JSON/YAML文件是可读的,可以手动执行根据需要编辑。

然后,您可以根据这些数据构建新的模型:
# model reconstruction from JSON:
from keras.models import model_from_json
model = model_from_json(json_string)

# model reconstruction from YAML
from keras.models import model_from_yaml
model = model_from_yaml(yaml_string)