点这里,搜索淘宝优惠券!
点这里,搜索淘宝优惠券!

16 模型持久化

【保存模型与参数】

1 my_model = create_model_function( ...... )
2 my_model.compile( ...... )
3 my_model.fit( ...... )
4 my_model. save( filepath, overwrite: bool=True, include_optimizer: bool=True )
5 filepath:保存的路径
6 overwrite:如果存在源文件,是否覆盖
7 include_optimizer:是否保存优化器状态
8 
9 示例 : mymodel.save(filepath="modelName.h5", includeoptimizer=False)

在Keras的官方应用中,加载Inception V3模型的内部实现为首先创建模型结构,然后利用模型的加载参数的方法加载参数。

 

【加载模型与参数】,这里直接创建模型,

1 my_model = keras . models . load_model( filepath ) 

 

【只保存参数/模型】

# 保存网络结构,载入网络结构
from keras.models import model_from_json
json_string = model.to_json()
model = model_from_json(json_string)
# 保存参数,载入参数
model.save_weights('my_model_weights.h5')
model.load_weights('my_model_weights.h5')

 

 

 

 

posted @ 2018-09-10 13:58  chd_ailex  阅读(165)  评论(0编辑  收藏  举报