機械学習・自然言語処理の勉強メモ

学んだことのメモやまとめ

modelの保存・ロード(Keras)

一度、fitで学習させたモデル(と重み)は下記の方法で保存できる。

model_json_str = model.to_json()
open('model.json', 'w').write(model_json_str)
model.save_weights('weight.h5')

ロードは下記の方法

from keras.models import model_from_json
model = model_from_json(open('model.json').read())
model.load_weights('weight.h5')