Kerasの学習モデルを保存する

2019年1月6日

AIの学習モデルは重要な資産です。

保存するとき

model.save('kabu_model.h5')

読み込むとき

from keras.models import model_from_json
from keras.models import load_model

# モデルと学習データの読み込み
model = load_model('kabu_model.h5')

# 実行
results = model.predict(np.array([[-100, 200, 100]]))

print("Predict:\n", results)

まとめ

拡張子がなぜh5なのかはよくわかりませんが、このようにすると学習データとモデルを保存して後で使いまわすことができます。現実問題として毎回学習というのは難しいので、事前に学習しておいたデータを使うということになりそうです。