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

时间:2019-06-04 16:07:23

标签: machine-learning keras deep-learning reinforcement-learning keras-rl

根据这些人(https://nihit.github.io/resources/spaceinvaders.pdf),可以通过深度强化学习进行早期停止。之前,我曾在Keras上进行过深度学习,但是在keras-rl上该怎么做?在相同的fit()函数中还是在将模型发送给代理之前?

1 个答案:

答案 0 :(得分:1)

看起来您可以只使用keras的回调;如果您确实希望将其放入包装中,请从here抓取并将其放入here。否则,我会尝试:

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)