A collection of machine learning algorithms for training in OpenAI Gym.
DQN (Mnih et al. 2015)
- Python
- OpenAI Gym
- TensorFlow
- NumPy
Run agent.py
and specify the environment and algorithm to use. For example:
python3 agent.py Breakout-v0 dqn
Optional settings can be viewed with
python3 agent.py -h