美文网首页深度学习-推荐系统-CV-NLP
keras 保存和加载权重及模型结构

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

作者: 春二一 | 来源:发表于2019-11-22 15:41 被阅读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")

相关文章

网友评论

    本文标题:keras 保存和加载权重及模型结构

    本文链接:https://www.haomeiwen.com/subject/pwwclctx.html