/cartpole-qlearning

A QLearning agent written in Python to balance an inverted pendulum

Primary LanguagePythonMIT LicenseMIT

cartpole-qlearning

The result of some playing around with Gym to train an agent to balance an inverted pendulum using Q-Learning.

The current code manages to solve the problem by only using the angle and the angular velocity of the pole, completely ignoring the linear position and velocity of the cart (to reduce dimensionality for faster convergence). I'm sure it could be tweaked even further to improve the results.