unfinity commited on
Commit
c8b11f3
1 Parent(s): 86410d7

Push Q-Learning agent to Hub

Browse files
Files changed (5) hide show
  1. .gitattributes +1 -0
  2. README.md +36 -0
  3. q-learning.pkl +3 -0
  4. replay.mp4 +3 -0
  5. results.json +1 -0
.gitattributes CHANGED
@@ -30,3 +30,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
30
  *.zip filter=lfs diff=lfs merge=lfs -text
31
  *.zst filter=lfs diff=lfs merge=lfs -text
32
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
30
  *.zip filter=lfs diff=lfs merge=lfs -text
31
  *.zst filter=lfs diff=lfs merge=lfs -text
32
  *tfevents* filter=lfs diff=lfs merge=lfs -text
33
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - FrozenLake-v1-8x8
4
+ - q-learning
5
+ - reinforcement-learning
6
+ - custom-implementation
7
+ model-index:
8
+ - name: q-FrozenLake-v1-8x8-200
9
+ results:
10
+ - task:
11
+ type: reinforcement-learning
12
+ name: reinforcement-learning
13
+ dataset:
14
+ name: FrozenLake-v1-8x8
15
+ type: FrozenLake-v1-8x8
16
+ metrics:
17
+ - type: mean_reward
18
+ value: 0.59 +/- 0.49
19
+ name: mean_reward
20
+ verified: false
21
+ ---
22
+
23
+ # **Q-Learning** Agent playing **FrozenLake-v1**
24
+ This is a trained model of a **Q-Learning** agent playing **FrozenLake-v1** .
25
+
26
+ ## Usage
27
+ ```python
28
+ model = load_from_hub(repo_id="unfinity/q-FrozenLake-v1-8x8-200", filename="q-learning.pkl")
29
+
30
+ # Don't forget to check if you need to add additional attributes (is_slippery=False 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:5dfa5156050bc4f663a761ac1c68a62f64c151b863ad1723a7702ea895ef6f09
3
+ size 2439
replay.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf58bb0e3da94f82e32f667500f55b952da427910bfdf65ef87aa0dc7804f4dd
3
+ size 152773
results.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"env_id": "FrozenLake-v1", "mean_reward": 0.586, "n_eval_episodes": 1000, "eval_datetime": "2022-09-06T05:02:00.720254"}