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'