必须先安装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')
一般来说,我们应在训练环境训练出模型,然后拿训练好的模型(即一个文件)到生产环境去部署。