/dqn

Deep Q-Network

Primary LanguagePython

Deep Q-Network

Deep Q-learning agent for the OpenAI Gym. Pre-trained models for the CartPole-v1 and MountainCar-v0 environments included. Includes a command line interface to train, run, and generate GIFs of an agent.

Installation

In order to generate GIFs of your agent running you must have ffmpeg installed. To install it type the following into a Python shell:

import imageio
imageio.plugins.ffmpeg.download()

Trained Agents

CartPole-v1

MountainCar-v0