Keras保存模型

Keras 保存模型

2018/6/3

用Keras训练好模型后,我们可以将其保存起来,以便以后用于使用或者在此基础上再次训练。
下面给出已测试过的保存及载入方法
首先在Python中安装h5py模块:
pip install h5py

保存:
model.save("model.h5)

载入:
model=load_model("model.h5")

注意在使用load_model函数时需要加入声明:from keras.models import load_model

完成。

猜你喜欢

转载自blog.csdn.net/weixin_39986952/article/details/80556966