Skip to content

Yet Another Deep Q-Learning Network Implementation

Notifications You must be signed in to change notification settings

tomstewart89/YADQN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YADQN

YADQN is yet another implementation of Deep Q-Networks (DQN) described in the Deepmind paper Playing Atari with Deep Reinforcement Learning.

In this repository, DQN is used to control the CartPole-v1 environment on the openai gym. To do that, a Q-function is implemented in Keras to map a state into the relative value of applying a LEFT or a RIGHT force to the cart. The Q-function is then trained by minimising the TD error from a set of S,A,R,S' tuples randomly sampled from an experience replay buffer. Lastly, an e-greedy policy selects an action for a given state to control the cart.

Results

On most runs, the algorithm was able to learn to balance the cart after a few hundred episodes:

Still learning:

64

Getting there...

343

Nailed it!

512

About

Yet Another Deep Q-Learning Network Implementation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages