Keras--保存model文件和载入model文件

保存keras的model文件和载入keras文件的方法有很多。keras中的模型主要包括model和weight两个部分。Json文件、Yaml文件、HDF5文件

保存model部分的主要方法：一是通过json文件

Json文件

[python]  view plain  copy

1. # serialize model to JSON  
2. model_json = model.to_json()  
3. with open("model.json", "w") as json_file:  
4.     json_file.write(model_json)  

Yaml文件

[python]  view plain  copy

1. # save as YAML  
2. yaml_string = model.to_yaml()  

保存权重的方法：通过保存权重（系数）

HDF5文件

[cpp]  view plain  copy

1. # serialize weights to HDF5  
2. model.save_weights("model.h5")  
3. print("Saved model to disk")  

同时保存model和权重的方式：

[python]  view plain  copy

1. from keras.models import load_model  
2.   
3. model.save('model_weight.h5')  # creates a HDF5 file 'my_model.h5'  

载入 model的方法
json&hdf5

[python]  view plain  copy

1. # load json and create model  
2. json_file = open('model.json', 'r')  
3. loaded_model_json = json_file.read()  
4. json_file.close()  
5. loaded_model = model_from_json(loaded_model_json  

[python]  view plain  copy

1. from keras.models import load_model  
2.   
3. model = load_model('model.h5')  

载入权重

[python]  view plain  copy

1. # load weights into new model  
2. loaded_model.load_weights("model.h5")  
3. print("Loaded model from disk")