深度学习-推荐系统-CV-NLP

keras 保存和加载权重及模型结构

2019-11-22  本文已影响0人  春二一

模型结构可以很容易地被保存和加载,如下:

保存为JSON字串 json_string= model. to_json()
保存为YAML字串 yaml_string=model.to_yaml() to_yam1()
从JsoN字串重构模

from keras.models import model_from_json
model= model_from_json(json_string)

从YAML字串重构模型

model= model_from_yaml(yaml_string)

模型参数(权重)也可以很容易地被保存和加载,如下

from keras.models import load_model model. save('my model. h5 ')

创建HDF5文件' my_model.h5
删除存在的模型
返回编译好的模型
和前一模型相同,等于load_model(' my_ model.h5")

上一篇下一篇

猜你喜欢

热点阅读