lab_laboratory

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

メモ 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の保存の時に使用したコードはこちら

with open('ts100model.pickle', mode='wb') as fp:
pickle.dump(model, fp)