Keras学习笔记（保存与加载）

1. 仅权重

model.save_weights(filepath) # 将模型权重存储为 HDF5 文件。
model.load_weights(filepath) # 加载模型

2. 仅结构

1. get_config()

config = model.get_config() # 模型配置信息的字典
model = Model.from_config(config)
# 或者，对于 Sequential:
model = Sequential.from_config(config)

2. to_json()

json_string = model.to_json() # 不包括权重，仅包含结构
model = model_from_json(json_string)

3. 权重和结构

model.save(filepath) # 保存
model.load_model(filepath) #加载模型

# 保存模型
model_json = model.to_json()
json_file = open("model.json", "w")
json_file.write(model_json)
json_file.close()
# 保存权重
model.save_weights("model.h5")

# 读取模型
from keras.models import model_from_json
json_file = open('model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)
# 读取权重
loaded_model.load_weights("model.h5")