The file reinforce_cartpole.py is composed of an agent (Neural Network) and the training of a model for the CartPole problem.
The file [reinforce_cartpole.py](https://gitlab.ec-lyon.fr/mghelfi/reinforcement-learning/-/blob/main/reinforce_cartpole.py) is composed of an agent (Neural Network) and the training of a model for the CartPole problem.
The graph of the evolution of the total rewards during the episodes is present in the file : image.png
<palign="center">
<imgsrc="image.png"width="350"title="hover text">
</p>
## Stable-Baselines3
The file a2c_sb3_cartpole.py contains a model to solve the CartPole problem using an Advantage Actor-Critic (A2C) algorithm with the Stable-Baselines3 library.
The file [a2c_sb3_cartpole.py](https://gitlab.ec-lyon.fr/mghelfi/reinforcement-learning/-/blob/main/a2c_sb3_cartpole.py) contains a model to solve the CartPole problem using an Advantage Actor-Critic (A2C) algorithm with the Stable-Baselines3 library.