Keras保存模型
in 机器学习 with 0 comment

Keras保存模型

in 机器学习 with 0 comment

将模型训练完了谁都不希望下次再使用的时候重新开始训练。
Keras当然提供了保存模型的方法。
通常情况下建议这么使用:

from keras.models import load_model
model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model
# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

这个方法保存了模型的所有的信息,包括结构,权重等配置信息。

Responses