2016-12-09 3 views
12

を節約するために、我々は歴史にmodel.fitの出力を返すことができます。keras:次のようにどのようにトレーニング履歴にKerasで

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

、更なる用途のためにファイルに歴史を保存する方法(例えばドローエポックに対するaccまたは損失のプロット)?

答えて

5

historyオブジェクトには、すべてのトレーニングエポックにまたがるさまざまなトレーニングメトリックを保持する辞書であるhistoryがあります。だからhistory.history['loss'][99]は、100回目の訓練でモデルの喪失を返します。これを保存するには、pickle辞書か、この辞書の別のリストを適切なファイルに保存するだけです。

13

私は何を使用すると、以下の通りである。

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

私は後で損失や精度をプロットしたい場合は辞書としての歴史を保存するこの方法で。

+0

ありがとうございます!私はちょうどこれをテストし、それは動作します。私はそれが正解であると言いたい。実際のHistoryオブジェクトには他の情報が含まれていますが、history.historyには必要なものがすべて含まれています。 – Bleyddyn

関連する問題