Upload folder using huggingface_hub
Browse files- README.md +9 -17
- hyperparameters.json +1 -0
- model.pt +3 -0
- replay.mp4 +0 -0
- results.json +1 -1
README.md
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
---
|
| 2 |
tags:
|
| 3 |
-
-
|
| 4 |
-
-
|
| 5 |
- reinforcement-learning
|
| 6 |
- custom-implementation
|
|
|
|
| 7 |
model-index:
|
| 8 |
- name: Taxi
|
| 9 |
results:
|
|
@@ -11,25 +12,16 @@ model-index:
|
|
| 11 |
type: reinforcement-learning
|
| 12 |
name: reinforcement-learning
|
| 13 |
dataset:
|
| 14 |
-
name:
|
| 15 |
-
type:
|
| 16 |
metrics:
|
| 17 |
- type: mean_reward
|
| 18 |
-
value:
|
| 19 |
name: mean_reward
|
| 20 |
verified: false
|
| 21 |
---
|
| 22 |
|
| 23 |
-
# **
|
| 24 |
-
This is a trained model of a **
|
| 25 |
-
|
| 26 |
-
## Usage
|
| 27 |
-
|
| 28 |
-
```python
|
| 29 |
-
|
| 30 |
-
model = load_from_hub(repo_id="Kedar1977/Taxi", filename="q-learning.pkl")
|
| 31 |
-
|
| 32 |
-
# Don't forget to check if you need to add additional attributes (is_slippery=False etc)
|
| 33 |
-
env = gym.make(model["env_id"])
|
| 34 |
-
```
|
| 35 |
|
|
|
|
| 1 |
---
|
| 2 |
tags:
|
| 3 |
+
- CartPole-v1
|
| 4 |
+
- reinforce
|
| 5 |
- reinforcement-learning
|
| 6 |
- custom-implementation
|
| 7 |
+
- deep-rl-class
|
| 8 |
model-index:
|
| 9 |
- name: Taxi
|
| 10 |
results:
|
|
|
|
| 12 |
type: reinforcement-learning
|
| 13 |
name: reinforcement-learning
|
| 14 |
dataset:
|
| 15 |
+
name: CartPole-v1
|
| 16 |
+
type: CartPole-v1
|
| 17 |
metrics:
|
| 18 |
- type: mean_reward
|
| 19 |
+
value: 500.00 +/- 0.00
|
| 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:807aef258bec4ebb89c02e6249687390e2a3b119e62b9f71f7a0dd37b537ab58
|
| 3 |
+
size 3264
|
replay.mp4
CHANGED
|
Binary files a/replay.mp4 and b/replay.mp4 differ
|
|
|
results.json
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
{"env_id": "
|
|
|
|
| 1 |
+
{"env_id": "CartPole-v1", "mean_reward": 500.0, "n_evaluation_episodes": 10, "eval_datetime": "2024-10-18T16:50:14.234553"}
|