Keras 模型持久化(保存,载入)

必须先安装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')

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

猜你喜欢

转载自my.oschina.net/jthmath/blog/1791841