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