Keras to_json(),它保存什么?

时间:2018-08-22 17:51:36

标签: python json keras

我的印象是,它仅保存了模型的体系结构,因此我应该能够在开始训练之前调用它?然后save_weights()保存恢复模型所需的权重吗?还有更多细节吗?

我可以在什么阶段致电to_json()?即我必须先打compile()吗?可以在fit()之前吗?

1 个答案:

答案 0 :(得分:1)

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)