CoolFace
Modelpublic

Punit71/firefighter-gridworld-leaderboard

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes
Model Card

๐Ÿš’ Firefighter GridWorld Leaderboard

A reinforcement learning benchmark in a 4x4 grid world where the agent must:

  1. 1.Pick up a water bucket ๐Ÿ’ง
  2. 2.Extinguish a fire ๐Ÿ”ฅ
  3. 3.Reach the goal ๐Ÿ

The environment features deterministic and stochastic versions with discrete actions, rewards, penalties, and sprite-based rendering.


๐Ÿ“Š Leaderboard (300 Episodes)

RankModelMean RewardStd DevSuccess RateNotes
๐Ÿฅ‡ 1MCTS27.43.641.0050 simulations, random rollout
2PPO4.05.83\~0.40Trained with Stable-Baselines3
3DQNโ€“30.023.9โŒFailed task consistently

๐Ÿงช Evaluation Protocol

  • โ€”Each agent is evaluated over 300 episodes
  • โ€”Maximum steps per episode: 60
  • โ€”Environment starts with the robot in the top-left
  • โ€”Rewards:
  • โ€”+10: extinguish fire and reach goal
  • โ€”โ€“1: step penalty
  • โ€”โ€“5: invalid actions or skipping steps

๐Ÿ›  Setup

bash
pip install -r requirements.txt

๐Ÿš€ Evaluate Your Agent

  1. 1.Clone the repo:
bash
git clone https://huggingface.co/spaces/YOUR_USERNAME/firefighter-gridworld-leaderboard
cd firefighter-gridworld-leaderboard
  1. 1.Run evaluation:
bash
python evaluation/evaluate_custom_agent.py --path ./my_agent.zip --algo PPO
  1. 1.Submit your eval_results.json via Pull Request.

๐Ÿง  Environment API

Custom environment follows Gymnasium standards:

python
import gymnasium as gym
from env.firefighter_env import FireFighterEnv

env = FireFighterEnv()
obs, info = env.reset()
for _ in range(60):
    action = env.action_space.sample()
    obs, reward, terminated, truncated, info = env.step(action)
    if terminated or truncated:
        break

๐Ÿ“ฅ Submissions

Include in your Pull Request:

  • โ€”eval_results.json
  • โ€”Description of your model and training setup
  • โ€”GIF of successful episode (optional)

๐Ÿ“ฆ Files

  • โ€”env/ โ€“ environment code
  • โ€”agents/ โ€“ training scripts (PPO, DQN, MCTS)
  • โ€”evaluation/ โ€“ evaluation and rendering
  • โ€”models/ โ€“ saved agents
  • โ€”assets/ โ€“ sprites and animation

๐Ÿ“œ License

MIT License. Contributions welcome!