Kedar1977 commited on
Commit
9884d2a
·
verified ·
1 Parent(s): 0b3d905

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. README.md +9 -17
  2. hyperparameters.json +1 -0
  3. model.pt +3 -0
  4. replay.mp4 +0 -0
  5. results.json +1 -1
README.md CHANGED
@@ -1,9 +1,10 @@
1
  ---
2
  tags:
3
- - Taxi-v3
4
- - q-learning
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: Taxi-v3
15
- type: Taxi-v3
16
  metrics:
17
  - type: mean_reward
18
- value: 7.56 +/- 2.71
19
  name: mean_reward
20
  verified: false
21
  ---
22
 
23
- # **Q-Learning** Agent playing1 **Taxi-v3**
24
- This is a trained model of a **Q-Learning** agent playing **Taxi-v3** .
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": "Taxi-v3", "mean_reward": 7.56, "n_eval_episodes": 100, "eval_datetime": "2024-10-18T15:30:24.529723"}
 
1
+ {"env_id": "CartPole-v1", "mean_reward": 500.0, "n_evaluation_episodes": 10, "eval_datetime": "2024-10-18T16:50:14.234553"}