Keras 的模型持久化

简介

模型的训练和使用应该是独立的, 所以应该有对模型的 序列化/反序列化 机制.
keras提供了友好的方法来做 save 和 load, 将模型保存到一个后缀为.h5的二进制文件中.

例子

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')

参考

官网FAQ:how-can-i-save-a-keras-model