Upload folder using huggingface_hub

README.md

@@ -0,0 +1,40 @@
+---
+tags:
+- CartPole-v1
+- deep-q-learning
+- reinforcement-learning
+- custom-implementation
+model-index:
+- name: dqn-cartpole
+  results:
+  - task:
+      type: reinforcement-learning
+      name: reinforcement-learning
+    dataset:
+      name: CartPole-v1
+      type: CartPole-v1
+    metrics:
+    - type: mean_reward
+      value: 483.30 +/- 50.10
+      name: mean_reward
+      verified: false
+---
+
+    # **Deep Q-Learning** Agent playing **CartPole-v1**
+    This is a trained model of a **Deep Q-Learning** agent playing **CartPole-v1**.
+    
+    ## Usage
+    
+    ```python
+    from huggingface_hub import hf_hub_download
+    import torch
+    from dqn_model import DQN
+    
+    model_path = hf_hub_download(repo_id="figurek1m/dqn-cartpole", filename="dqn_cartpole.pth")
+    q_network = DQN(input_dim=4, output_dim=2)
+    q_network.load_state_dict(torch.load(model_path))
+    q_network.eval()
+    
+    env = gym.make("CartPole-v1")
+    ```
+    

dqn_cartpole.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f4e5202161d071fa3438edf276a857303818d6acb86920d4534693bda185137
+size 273018

results.json

@@ -0,0 +1 @@
+{"env_id": "CartPole-v1", "mean_reward": 483.3, "n_eval_episodes": 10, "eval_datetime": "2025-03-03T13:55:30.195731"}