nikitakapitan commited on
Commit
f10a3e9
1 Parent(s): a9d3979

Push Q-Learning agent to Hub

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