保存Keras训练的模型

保存Keras训练的模型

不推荐使用pickle或cPickle。

(1) 如果只保存模型结构，代码如下：

[python]  view plain  copy

1. # save as JSON  
2. json_string = model.to_json()  
3. # save as YAML  
4. yaml_string = model.to_yaml()  
5. # model reconstruction from JSON:  
6. from keras.modelsimport model_from_json  
7. model = model_from_json(json_string)  
8.    
9. # model reconstruction from YAML  
10. model =model_from_yaml(yaml_string)  

(2) 如果需要保存数据：

[python]  view plain  copy

1. model.save_weights('my_model_weights.h5')  
2. model.load_weights('my_model_weights.h5')  

(3) 综合运用：

[python]  view plain  copy

1. json_string = model.to_json()  
2. open('my_model_architecture.json','w').write(json_string)  
3. model.save_weights('my_model_weights.h5')  
4.    
5. model = model_from_json(open('my_model_architecture.json').read())  
6. model.load_weights('my_model_weights.h5')