技術系メモ

メモ。内容は保証しません。

【keras】keras のモデルの保存と読み込み

モデルの保存

# モデルの保存
json_model = model.to_json()
with open("model.json", "w") as f
  json.dump(text, f)

# パラメータの保存
model.save_weights("param.hdf5")

保存したモデルの読み込み

from keras.models import model_from_json

# モデルの読み込み
json_model = json.load(open("model.json"))
model = model_from_json(json_model)

# パラメータの読み込み
model.load_weights("param.hdf5")