Downloads · 30 days
0
tms-marl-gnn-models
Kunalsinghh/tms-marl-gnn-models
tms-marl-gnn-models is a reinforcement learning model from Kunalsinghh. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Coordinated multi-intersection traffic control using MARL algorithms.
Access
Public
Updated Dec 24, 2025
Repo size
4.2 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt7.1 MB · 91%
At a glance
- Task
- Reinforcement Learning
- License
- apache-2.0
- Access
- Public
- Created
- Dec 24, 2025
- Updated
- Dec 24, 2025
- SHA
- aae90ed7
Model card
From the Hugging Face model README
TMS2 - MARL Traffic Management Models
Multi-Agent Reinforcement Learning Models
Coordinated multi-intersection traffic control using MARL algorithms.
Algorithms:
- IQL: Independent Q-Learning
- VDN: Value Decomposition Networks
- QMIX: Monotonic value factorization
Features:
- Multi-intersection coordination
- Green wave optimization
- Network-wide throughput maximization
Model Description
These models are part of the Traffic Management System 2 (TMS2) project, an intelligent traffic control system using deep learning and reinforcement learning.
Training Details
- Framework: PyTorch
- Training Platform: Google Colab (T4 GPU)
- Training Date: December 2025
Usage
import torch
# Load model
model = torch.load('model.pt')
model.eval()
# Inference
with torch.no_grad():
output = model(input_tensor)
License
Apache 2.0
- Task
- Reinforcement Learning
- License
- apache-2.0
- Created
- Dec 24, 2025
- Updated
- Dec 24, 2025