首页 > 其他 > 详细

keras保存和载入模型

时间:2020-05-20 11:36:33      阅读:62      评论:0      收藏:0      [点我收藏+]

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( preds[1] ) )
————————————————
原文链接:https://blog.csdn.net/wslkd0123/article/details/80647041

keras保存和载入模型

原文:https://www.cnblogs.com/qingchen-forever/p/12922307.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!