Keras 模型持久化（保存，载入）

2019独角兽企业重金招聘Python工程师标准>>>

必须先安装h5py：

pip3 install h5py

 

保存模型

# 假设 model 是训练完毕的模型
# 将模型保存到文件 my_model.h5
model.save('my_model.h5')

 

载入模型

# 要载入模型，需要导入此函数
from keras.models import load_model

# 从文件 my_model.h5 中载入模型
model = load_model('my_model.h5')

 

一般来说，我们应在训练环境训练出模型，然后拿训练好的模型（即一个文件）到生产环境去部署。

转载于:https://my.oschina.net/jthmath/blog/1791841