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")