参考链接:
模型保存与加载:
保存:
#保存模型
model.save('model.h5')
# 保存参数
model.save_weights('my_model_weights.h5')
加载:
from keras.models import load_model
#载入模型、参数
model = load_model('model.h5')
model.load_weights('my_model_weights.h5')
# 保存网络结构,载入网络结构
from keras.models import model_from_json
json_string = model.to_json()
model = model_from_json(json_string)
print(json_string)