keras_dqn.py 文件源码

def save_queue(EXPERIENCE_MEMORY):
    with open('saved nets/saved_queue_new.pkl','wb') as f:
        cPickle.dump(EXPERIENCE_MEMORY,f,protocol=cPickle.HIGHEST_PROTOCOL)

    call(['rm','saved nets/saved_queue.pkl'])
    call(['mv','saved nets/saved_queue_new.pkl','saved nets/saved_queue.pkl'])