Please don't revise test.py, environment.py, agent_dir/agent.py
Type the following command to install OpenAI Gym Atari environment.
$ pip3 install opencv-python gym gym[atari]
Please refer to OpenAI's page if you have any problem while installing.
training policy gradient:
$ python3 main.py --train_pg
testing policy gradient:
$ python3 test.py --test_pg
training DQN:
$ python3 main.py --train_dqn
testing DQN:
$ python3 test.py --test_dqn