/dqn-learning-from-data

Primary LanguagePython

dqn-learning-from-data

This repo implements Deep Q-learning from DQN (Mnih et al.).

The repository has been forked and inspired by https://github.com/danielegrattarola/deep-q-atari

Setup

Setup your virtual environment of choice.

Install the following dependencies:

  • keras
  • gym
  • gym[atari]
  • tensorflow
  • numpy

Usage

Train the neural network:

python atari.py --train

for help use:

python atari.py -h