You need to install OpenAI Gym
、OpenAI Gym Atari
、Gym Box2D
environment.
Please refer to OpenAI's page if you have any problem while installing.
training policy gradient:
python main.py --train_pg
training DQN:
python main.py --train_dqn
PS: You need to load model which has best performance before testing the algorithm
testing policy gradient:
python test.py --test_pg
testing DQN:
python test.py --test_dqn