lab_laboratory

情報学生のアウトプットを行っていくためのブログ

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)