Обучение глубокому усилению (keras-rl) Ранняя остановка

Question

По словам этих ребят (https://nihit.github.io/resources/spaceinvaders.pdf), можно выполнить Ранний останов с глубоким обучением подкреплению. Я использовал это раньше с Deep Learning на 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)