ThomasSimonini HF staff commited on
Commit
d96c7ad
1 Parent(s): 7d44a67

Push Q-Learning agent to Hub

Browse files
Files changed (5) hide show
  1. .gitattributes +2 -0
  2. README.md +36 -0
  3. q-learning.pkl +3 -0
  4. replay.mp4 +3 -0
  5. results.json +1 -0
.gitattributes CHANGED
@@ -25,3 +25,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
29
+ q-learning.pkl filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - Taxi-v3
4
+ - q-learning
5
+ - reinforcement-learning
6
+ - custom-implementation
7
+ model-index:
8
+ - name: q-Taxi-v3
9
+ results:
10
+ - metrics:
11
+ - type: mean_reward
12
+ value: 7.56 +/- 2.71
13
+ name: mean_reward
14
+ task:
15
+ type: reinforcement-learning
16
+ name: reinforcement-learning
17
+ dataset:
18
+ name: Taxi-v3
19
+ type: Taxi-v3
20
+ ---
21
+
22
+ # **Q-Learning** Agent playing **Taxi-v3**
23
+ This is a trained model of a **Q-Learning** agent playing **Taxi-v3** .
24
+
25
+ ## Usage
26
+ ```
27
+
28
+ pickle_model = load_from_hub(repo_id="ThomasSimonini/q-Taxi-v3", filename="q-learning.pkl")
29
+
30
+ # Don't forget to check if you need to add additional attributes (is_slippery etc)
31
+ env = gym.make(model["env_id"])
32
+
33
+ evaluate_agent(env, model["max_steps"], model["n_eval_episodes"], model["qtable"], model["eval_seed"])
34
+
35
+ ```
36
+
q-learning.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70a9d2b80100a3d2a021b024f8dbbdfdcde5cd91748db86610836ce7c388f159
3
+ size 24649
replay.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4cfb5953d02d3e25681530a7a86ca79a704a8baa9b3a4a04be2c6005af381b2
3
+ size 101069
results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"env_id": "Taxi-v3", "mean_reward": 7.56, "n_eval_episodes": 100, "eval_datetime": "2022-05-19T07:24:32.507777"}