Skip to content

Latest commit

 

History

History
19 lines (10 loc) · 1.15 KB

README.md

File metadata and controls

19 lines (10 loc) · 1.15 KB

Machine Learning Nano Degree Capstone Project: Deep Reinforcement Learning to train an agent to play Atari Pong game using DQN

Random model DQN 5260000 steps DQN 8090000 steps DQN 9500000 steps
Random model DQN 5260000 steps DQN 8090000 steps DQN 9500000 steps

I have obtained the starter code from CS 294: Deep Reinforcement Learning, Spring 2017

https://github.com/berkeleydeeprlcourse/homework/tree/master/sp17_hw/hw3

Make sure to install Tensorflow, OpenAI Gym, numpy, pandas and bokeh before running these scripts.

Run the training using the following command

python run_dqn_atari.py >> train_logs_new.log

There is a jupyter notebook, named DQN report.ipynb, which has all the code for plotting the stats presented in the report. The file named DQN_report.html is the html snapshot of the jupyter notebook. The file named record.pdf has the complete report of the project.