keras模型保存与加载

本文内容来自于个人理解和网上参考内容的综合，难免有错误（如能指正 不胜感激），仅供参考。

参考https://blog.csdn.net/tszupup/article/details/85198949

https://blog.csdn.net/zengNLP/article/details/94589469

1、模型保存及加载

1.1同时保存模型及参数

通过下面的方式加载

# 加载整个模型
model = load_model('my_model.h5')

 

1.2分开保存模型及参数之  结构保存

我们使用to_json()方法或者to_yaml()将模型结构保存到json文件或者yaml文件。
 

# 保存模型的结构
json_string = model.to_json()        # 方式1
open('model_architecture_1.json', 'w').write(json_string)
yaml_string = model.to_yaml()        # 方式2
open('model_arthitecture_2.yaml', 'w').write(yaml_st