Upload STGFORMER_MAMBA_FAST model trained on PEMS-BAY

README.md

@@ -0,0 +1,57 @@
+---
+tags:
+- traffic-forecasting
+- time-series
+- graph-neural-network
+- stgformer_mamba_fast
+datasets:
+- pems-bay
+---
+
+# Spatial-Temporal Graph Transformer (Mamba Fast) - PEMS-BAY
+
+Spatial-Temporal Graph Transformer (Mamba Fast) (STGFORMER_MAMBA_FAST) trained on PEMS-BAY dataset for traffic speed forecasting.
+
+## Model Description
+
+STGFormer with optimized Mamba SSM (fast config for validation)
+
+
+
+## Dataset
+
+**PEMS-BAY**: Traffic speed data from highway sensors.
+
+## Usage
+
+```python
+from utils.stgformer import load_from_hub
+
+# Load model from Hub
+model, scaler = load_from_hub("PEMS-BAY", hf_repo_prefix="STGFORMER_MAMBA_FAST")
+
+# Get predictions
+from utils.stgformer import get_predictions
+predictions = get_predictions(model, scaler, test_dataset)
+```
+
+## Training
+
+Model was trained using the STGFORMER_MAMBA_FAST implementation with default hyperparameters.
+
+## Citation
+
+If you use this model, please cite the original STGFORMER_MAMBA_FAST paper:
+
+```bibtex
+@inproceedings{lan2022stgformer,
+  title={STGformer: Spatial-Temporal Graph Transformer for Traffic Forecasting},
+  author={Lan, Shengnan and Ma, Yong and Huang, Weijia and Wang, Wanwei and Yang, Hui and Li, Peng},
+  booktitle={IEEE Transactions on Neural Networks and Learning Systems},
+  year={2022}
+}
+```
+
+## License
+
+This model checkpoint is released under the same license as the training code.

config.json

@@ -0,0 +1,28 @@
+{
+  "num_nodes": 325,
+  "in_steps": 12,
+  "out_steps": 12,
+  "input_dim": 1,
+  "output_dim": 1,
+  "steps_per_day": 288,
+  "input_embedding_dim": 24,
+  "tod_embedding_dim": 24,
+  "dow_embedding_dim": 0,
+  "spatial_embedding_dim": 0,
+  "adaptive_embedding_dim": 80,
+  "num_heads": 4,
+  "num_layers": 3,
+  "dropout_a": 0.3,
+  "use_mixed_proj": true,
+  "model_dim": 128,
+  "dataset": "PEMS-BAY",
+  "graph_mode": "learned",
+  "lambda_hybrid": 0.5,
+  "sparsity_k": null,
+  "propagation_mode": "power",
+  "temporal_mode": "mamba",
+  "mamba_d_state": 8,
+  "mamba_d_conv": 2,
+  "mamba_expand": 1,
+  "use_zero_init": true
+}

hub_metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "PEMS-BAY",
+  "metrics": {},
+  "framework": "PyTorch",
+  "hf_repo_prefix": "STGFORMER_MAMBA_FAST",
+  "implementation": "internal"
+}

metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "PEMS-BAY",
+  "upload_date": "2025-12-07T11:44:49.636535",
+  "metrics": {},
+  "framework": "PyTorch",
+  "model_type": "STGFORMER_MAMBA_FAST"
+}

model.safetensors

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