1. 程式人生 > >深度學習【1】ubuntu中利用h5py儲存訓練好的keras 神經網路模型

深度學習【1】ubuntu中利用h5py儲存訓練好的keras 神經網路模型

利用h5py儲存的模型所佔的空間非常小。在利用h5py儲存keras訓練好的模型之前需要先安裝h5py,具體的安裝過程可參考我關於h5py安裝的博文:http://blog.csdn.net/linmingan/article/details/50736300

利用h5py儲存和讀取keras模型的程式碼如下:

import h5py from keras.models import model_from_json
json_string = model.to_json()
open('my_model_architecture.json','w').write(json_string)
model.save_weights('my_model_weights.h5')
 #讀取model
model = model_from_json(open('my_model_architecture.json').read())
model.load_weights('my_model_weights.h5')