franckhu's picture
Update README.md
00cb826 verified
|
raw
history blame contribute delete
No virus
839 Bytes
---
library_name: stable-baselines3
tags:
- CartPole-v1
- deep-reinforcement-learning
- reinforcement-learning
- stable-baselines3
model-index:
- name: A2C
results:
- task:
type: reinforcement-learning
name: reinforcement-learning
dataset:
name: CartPole-v1
type: CartPole-v1
metrics:
- type: mean_reward
value: 147.40 +/- 15.48
name: mean_reward
verified: false
---
# **A2C** Agent playing **CartPole-v1**
This is a trained model of a **A2C** agent playing **CartPole-v1**
using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
```python
from stable_baselines3 import A2C
from huggingface_sb3 import load_from_hub
checkpoint = load_from_hub(
repo_id="franckhu/apprentissage_auto_CartPole-v1-a2c",
filename="CartPole-v1-a2c.zip",
)
...
```