gabrielbenabou
commited on
Commit
•
07dffb5
1
Parent(s):
a422e34
Upload folder using huggingface_hub
Browse files- README.md +9 -19
- hyperparameters.json +1 -0
- model.pt +3 -0
- results.json +1 -1
README.md
CHANGED
@@ -1,12 +1,12 @@
|
|
1 |
---
|
2 |
-
library_name: stable-baselines3
|
3 |
tags:
|
4 |
- CartPole-v1
|
5 |
-
-
|
6 |
- reinforcement-learning
|
7 |
-
-
|
|
|
8 |
model-index:
|
9 |
-
- name:
|
10 |
results:
|
11 |
- task:
|
12 |
type: reinforcement-learning
|
@@ -16,22 +16,12 @@ model-index:
|
|
16 |
type: CartPole-v1
|
17 |
metrics:
|
18 |
- type: mean_reward
|
19 |
-
value:
|
20 |
name: mean_reward
|
21 |
verified: false
|
22 |
---
|
23 |
|
24 |
-
# **
|
25 |
-
This is a trained model of a **
|
26 |
-
|
27 |
-
|
28 |
-
## Usage (with Stable-baselines3)
|
29 |
-
TODO: Add your code
|
30 |
-
|
31 |
-
|
32 |
-
```python
|
33 |
-
from stable_baselines3 import ...
|
34 |
-
from huggingface_sb3 import load_from_hub
|
35 |
-
|
36 |
-
...
|
37 |
-
```
|
|
|
1 |
---
|
|
|
2 |
tags:
|
3 |
- CartPole-v1
|
4 |
+
- reinforce
|
5 |
- reinforcement-learning
|
6 |
+
- custom-implementation
|
7 |
+
- deep-rl-class
|
8 |
model-index:
|
9 |
+
- name: Reinforce-CartPole-v1
|
10 |
results:
|
11 |
- task:
|
12 |
type: reinforcement-learning
|
|
|
16 |
type: CartPole-v1
|
17 |
metrics:
|
18 |
- type: mean_reward
|
19 |
+
value: 20.60 +/- 6.71
|
20 |
name: mean_reward
|
21 |
verified: false
|
22 |
---
|
23 |
|
24 |
+
# **Reinforce** Agent playing **CartPole-v1**
|
25 |
+
This is a trained model of a **Reinforce** agent playing **CartPole-v1** .
|
26 |
+
To learn to use this model and train yours check Unit 4 of the Deep Reinforcement Learning Course: https://huggingface.co/deep-rl-course/unit4/introduction
|
27 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
hyperparameters.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"h_size": 16, "n_training_episodes": 1000, "n_evaluation_episodes": 10, "max_t": 1000, "gamma": 1.0, "lr": 0.01, "env_id": "CartPole-v1", "state_space": 4, "action_space": 2}
|
model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b207304122a32079f5313add683aa92a29888c3fe12769db6cb7db1f341716d7
|
3 |
+
size 2579
|
results.json
CHANGED
@@ -1 +1 @@
|
|
1 |
-
{"
|
|
|
1 |
+
{"env_id": "CartPole-v1", "mean_reward": 20.6, "n_evaluation_episodes": 10, "eval_datetime": "2024-02-20T09:57:02.663013"}
|