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