mklissa / DQN

Implementation of DQN and DDQN in Theano

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DQN

This is an implementation of the classic DQN algorithm using Theano.

Requirements

  • Python 3.6
  • Theano 0.8.2
  • OpenCV
  • OpenAI Gym
  • Numpy

Trained weights are provided for the following games: Boxing, Breakout, Pong and SpaceInvaders.

To train the model, please use:

python DQN.py --env [environment name]

To see it play:

python play.py --env [environment name]

The pickle files must be named after the game (i.e. breakout.pkl). You can also change that directly in the script. As mentionned in this blog from OpenAI, DQN, as well as more recent algorithms, are subject to adversarial attacks. To see the effects on this implementation, choose:

python play.py --env [environment name] --adv 1

Results

About

Implementation of DQN and DDQN in Theano


Languages

Language:Python 100.0%