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") | |
``` | |