A Tensorflow DQN implementation based on DeepMind's DQN for playing Atari games.This the code for 'Build a game AI' on Youtube
- gym
- Tensorflow
- [virtualenv] (https://virtualenv.pypa.io/en/latest/installation.html)
To run, type the following into terminal
python atari.py --game <env_name>
It will run SpaceInvaders-v0
by default but you can use other game names as well. Add --display true
to the above command line argument if you'd like to see the game while it trains
Credit for the vast majority of code here goes to Kee Hyun Won. I've merely created a wrapper to get people started.