This repository contains code for solving brachistochronne problemusing reinforcement learning. A detailed post about the project can be found at : - https://medium.com/@aaryanshmohanbansal/solving-brachistochrone-problem-by-reinforcement-learning-16d7d2af2fbe
Requirements:
- Stable-Baselines3
- Tensorboard
- gym
- numpy
- matplotlib
- scipy
In the directory "rl_algorithms" , run python3 barhcistrone_ddpg.py
to start training for RL agent.
In the directory "rl_algorithms", run python3 run_best_model.py
to see the plot of best trajectory learnt by RL agent.