保存keras的model文件和载入keras文件的方法有很多。现在分别列出,以便后面查询。
keras中的模型主要包括model和weight两个部分。
保存model部分的主要方法:一是通过json文件
Json文件
# serialize model to JSON
model_json = model.to_json()
with open("model.json", "w") as json_file:
json_file.write(model_json)
Yaml文件
# save as YAML
yaml_string = model.to_yaml()
保存权重的方法:通过保存权重(系数)
HDF5文件
# serialize weights to HDF5
model.save_weights("model.h5")
print("Saved model to disk")
同时保存model和权重的方式:
from keras.models import load_model
model.save('model_weight.h5') # creates a HDF5 file 'my_model.h5'
载入model的方法
json&hdf5
# load json and create model
json_file = open('model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json
from keras.models import load_model
model = load_model('model.h5')
载入权重
# load weights into new model
loaded_model.load_weights("model.h5")
print("Loaded model from disk")