誰かがコールバックを使用してKerasの簡単な例を投稿して、すべてのエポックの後でモデルを保存できますか?重みを節約する例を見つけることができますが、すべてのトレーニングエポック後に完全に機能するモデルを保存できるようにしたいと考えています。
誰かがコールバックを使用してKerasの簡単な例を投稿して、すべてのエポックの後でモデルを保存できますか?重みを節約する例を見つけることができますが、すべてのトレーニングエポック後に完全に機能するモデルを保存できるようにしたいと考えています。
回答:
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)
改善されたモデルのみの保存や保存されたモデルのロードなど、いくつかの例がここにあります。
ファイルパスにエポック変数を必ず含めてください。そうでない場合、保存されたモデルはすべてのエポックの後で置き換えられます。
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')
その他の例については、こちらをご覧ください。