Keras使用HDF5文件系统来保存模型。模型操作方法很容易。
keras实现特别简单,直接调用save方法
mode_file = "xxx_model.h5"
#如果网络不存在备份文件,创建一个保存,路径可以自己定义
if not os.path.exists('xxx_model.h5'):
print('=======================保存网络模型=======================')
model.save('xxx_model.h5')
加载
使用前需要导入load_model方法。
from keras.models import load_model
#判断有没有网络文件,路径可以自己定义
mode_file = "xxx_model.h5"
if os.path.exists(mode_file ):
print('=======================加载网络模型=======================')
model = load_model(mode_file )