keras保存和加载模型的方法

keras模型主要分为model和weight两个部分,前者保存整个模型结构,后者仅保存权值

保存模型

仅保存权重

HDF5文件

model.save_weights("model.h5") 

保存整个模型及结构

model.save('model_weight.h5')

加载模型

载入权重

from keras.models import load_model  
loaded_model.load_weights("model.h5")  

载入整个模型结构

from keras.models import load_model
model = load_model('model.h5')  

tips:载入整个模型结构时,若模型训练时有自定义loss或metrics,则载入时会报类似错:Unknown metric function:my_loss (此处my_loss是一个自定义函数),则加载模型时需要指定custom_objects参数:

model = load_model('model.h5',{'my_loss': my_loss})
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页