1. ホーム
  2. python

[解決済み] keras: 履歴オブジェクトの学習履歴属性を保存する方法

2023-01-08 23:35:48

質問

Kerasでは、出力に model.fit を以下のように履歴に残します。

 history = model.fit(X_train, y_train, 
                     batch_size=batch_size, 
                     nb_epoch=nb_epoch,
                     validation_data=(X_test, y_test))

さて、historyオブジェクトのhistory属性をファイルに保存して、さらに利用する方法(例えば、エポックに対するaccやlossのプロットを描くなど)ですが、どのようにすればよいのでしょうか?

どのように解決するのですか?

私が使用しているのは、以下のものです。

    with open('/trainHistoryDict', 'wb') as file_pi:
        pickle.dump(history.history, file_pi)

このようにして履歴を辞書として保存しておき、後で損失や精度をプロットしたい場合に備えています。