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