top of page

Model Based Reinforcement Learning for Cart-pole Swing-up

  • Implemented Model Predictive Controller to balance the cartpole, the MPC has two components, the first tries to learn the environment dynamics and the other plans the policy (sequential actions) based on the current state 

  • Cross Entroy Maximization algorithm was used for the planner, the algorithm iteratively optimizes a set of Gaussian distributions that model a distribution over each action in a trajectory 

The second video shows the performance of the agent after training, the first video is a random policy in action

bottom of page