Push Q-Learning agent to Hub
Browse files- .gitattributes +2 -0
- README.md +35 -0
- q-learning.pkl +3 -0
- replay.mp4 +3 -0
- 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,35 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 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 | 
            +
              ```python
         | 
| 27 | 
            +
              model = load_from_hub(repo_id="btsas/q-Taxi-v3", 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:ce3cb01f56bf025e6485e2232f4ed6c705933419c10b386398e5e9fb4606c205
         | 
| 3 | 
            +
            size 24589
         | 
    	
        replay.mp4
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:33bae69b4f3af299fa207c1bbdb9deb96db34590ef9da7fc86c79950531c2847
         | 
| 3 | 
            +
            size 105500
         | 
    	
        results.json
    ADDED
    
    | @@ -0,0 +1 @@ | |
|  | 
|  | |
| 1 | 
            +
            {"env_id": "Taxi-v3", "mean_reward": 7.56, "n_eval_episodes": 100, "eval_datetime": "2022-05-20T21:47:35.565417"}
         |