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.
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()