好不容易訓練好的模型,可別一關閉程式就消失了,儲存、讀取的做法十分簡單
儲存模型:
將訓練好的模型存起來,可以將來繼續訓練、或是讀取使用
存成HDF5檔案, 如果沒有這個套件的話,請pip install h5pymodel.save('./model.h5')
讀取模型:model = load_model('./model.h5')
在Heroku伺服器,我們用的就是讀取已經訓練好的模型,省去訓練的時間
def init_default(self):
self.model = Sequential()
"""
建構一樣的模型...
"""
del self.model # deletes the existing model
self.model = load_model('./model/number_model.h5')
self.graph = tf.get_default_graph()
print('Reloaded')