Keras 保存模型
用Keras训练好模型后,我们可以将其保存起来,以便以后用于使用或者在此基础上再次训练。
下面给出已测试过的保存及载入方法
首先在Python中安装h5py模块:
pip install h5py
保存:
model.save("model.h5)
载入:
model=load_model("model.h5")
注意在使用load_model
函数时需要加入声明:from keras.models import load_model
完成。