すべてのエポックの後でモデルを保存するためのKerasコールバックの例?


17

誰かがコールバックを使用してKerasの簡単な例を投稿して、すべてのエポックの後でモデルを保存できますか?重みを節約する例を見つけることができますが、すべてのトレーニングエポック後に完全に機能するモデルを保存できるようにしたいと考えています。

回答:


21

Keras コールバック「ModelCheckpoint」で「save_weights_only」をFalseに設定すると、モデル全体が保存されます。上記のリンクから取得したこの例では、パフォーマンスに関係なく、すべてのエポックで完全なモデルを保存します。

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

改善されたモデルのみの保存や保存されたモデルのロードなど、いくつかの例がここにあります。


9

ファイルパスにエポック変数を必ず含めてください。そうでない場合、保存されたモデルはすべてのエポックの後で置き換えられます。

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

その他の例については、こちらをご覧ください


2
サイトへようこそ!おかげで、答えへの追加に感謝します
I_Play_With_Data
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.