本文介绍了使用Keras框架保存和加载模型的方法，包括将整个模型保存为JSON或YAML格式，仅保存和加载模型权重等操作。这些技巧对于模型训练后的持久化存储及后续部署十分重要。
*将模型保存为json
json_string = model.to_json()
将模型保存为yaml
yaml_string = model.to_yaml()
从保存的json中加载模型
from keras.modelsimport model_from_json
model = model_from_json(json_string)
从保存的yaml中加载模型
model =model_from_yaml(yaml_string)
保存与加载权重数据
model.save_weights(‘my_model_weights.h5’)
model.load_weights(‘my_model_weights.h5’)*
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
