dqn-cartpole / README.md
figurek1m's picture
Upload folder using huggingface_hub
d35ce9b verified
metadata
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: 500.00 +/- 0.00
            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")
```