pythonメモ CNNのモデル保存方法
↓が保存方法
ニューラルネットワークは「モデルの構成」と「学習済みの重み」の2つを学習時に必要とする.
# モデルの保存
open('and.json',"w").write(model.to_json())
# 学習済みの重みを保存
model.save_weights('and.h5')
# モデルの読み込み
model = model_from_json(open('and.json', 'r').read())
# 重みの読み込み
model.load_weights('and_weight.hdf5')
SVMでの保存方法は学習結果のもでるだったけど,ニューラルネットワークはそうなるのかって思った・・・勉強になるなぁ
ちなみに,SVMの保存の時に使用したコードはこちら
#modelの保存
with open('ts100model.pickle', mode='wb') as fp:
pickle.dump(model, fp)
#modelの読み込み
readmodel='SVMdatasetMIX8.pickle'
with open(readmodel, mode='rb') as fp:
model = pickle.load(fp)