PythonMania

普段はロボットとAIを組み合わせて色々作ってます。Python関係以外も色々投稿していくと思います。

【Python】Kerasで学習したモデルとパラメータの保存・読み込み方法 【DeepLearning】

f:id:mizuhiki0111:20190430180019p:plain



自分用にメモ




保存時

#モデルの保存
model_json_str =  model.to_json()
open(' model.json', 'w').write(model_json_str)

#パラメータの保存
model.save_weights(' model.h5');

読み込み時

#モデルの読み込み
model = model_from_json(open("~/model.json", 'r').read())

# 重みの読み込み
model.load_weights("~model.h5")