Keras如何保存和载入训练好的模型参数
发布时间
阅读量:
阅读量
1,模型保存:
my_model = create_model_function( ...... )
my_model.compile( ...... )
my_model.fit( ...... )
model_name . save( filepath, overwrite: bool=True, include_optimizer: bool=True )
filepath:模型存储的具体路径
overwrite:当目标文件已存在时,是否允许覆盖原有内容
include_optimizer:决定是否将优化器相关参数一同保存
ex : mymodel.save(filepath="p402/my_model.h5", includeoptimizer=False)
2,模型加载:
my_model = keras . models . load_model( filepath )
完成加载后可继续进行训练过程:
my_model . fit( X_train_2,Y_train_2 )
亦可直接进行性能评估:
preds = my_model . evaluate( X_test, Y_test )
print ( "Loss = " + str( preds[0] ) )
print ( "Test Accuracy = " + str( pr
全部评论 (0)
还没有任何评论哟~
