machine-learning - 深度强化学习 (keras-rl) 提前停止

Question

根据这些家伙 ( https://nihit.github.io/resources/spaceinvaders.pdf ) 的说法，可以使用深度强化学习执行 Early Stopping。我之前在 Keras 上使用过深度学习，但是，如何在 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)