machine-learning - 深層強化学習 (keras-rl) 早期停止

Question

これらの人 ( https://nihit.github.io/resources/spaceinvaders.pdf ) によると、深層強化学習で早期停止を実行することが可能です。以前、Keras の Deep Learning でそれを使用しましたが、keras-rl でそれを行うにはどうすればよいですか? 同じ fit() 関数で、またはモデルをエージェントに送信する前に?

Answer 1

keras のコールバックを使用できるようです。本当にパッケージに入れたい場合は、ここから取得して、ここに入れます。そうでなければ、私は試してみます：

from keras.callbacks import EarlyStopping

early_stop = EarlyStopping(patience=69) # epochs stagnation before termination

# from their example cem_cartpole.py
cem.fit(env, nb_steps=100000, visualize=False, callbacks=[early_stop], verbose=2)