Upload STGFORMER_FINAL model trained on METR-LA

README.md

@@ -0,0 +1,57 @@
+---
+tags:
+- traffic-forecasting
+- time-series
+- graph-neural-network
+- stgformer_final
+datasets:
+- metr-la
+---
+
+# Spatial-Temporal Graph Transformer (Final) - METR-LA
+
+Spatial-Temporal Graph Transformer (Final) (STGFORMER_FINAL) trained on METR-LA dataset for traffic speed forecasting.
+
+## Model Description
+
+STGFormer Chebyshev+TCN with Xavier initialization, DOW embeddings, exclude_missing_from_norm, and sparsity_k=16 [FINAL - 100 epochs]
+
+
+
+## Dataset
+
+**METR-LA**: Traffic speed data from highway sensors.
+
+## Usage
+
+```python
+from utils.stgformer import load_from_hub
+
+# Load model from Hub
+model, scaler = load_from_hub("METR-LA", hf_repo_prefix="STGFORMER_FINAL")
+
+# Get predictions
+from utils.stgformer import get_predictions
+predictions = get_predictions(model, scaler, test_dataset)
+```
+
+## Training
+
+Model was trained using the STGFORMER_FINAL implementation with default hyperparameters.
+
+## Citation
+
+If you use this model, please cite the original STGFORMER_FINAL 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,33 @@
+{
+  "num_nodes": 207,
+  "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": 24,
+  "spatial_embedding_dim": 0,
+  "adaptive_embedding_dim": 80,
+  "num_heads": 4,
+  "num_layers": 3,
+  "dropout_a": 0.3,
+  "use_mixed_proj": true,
+  "pre_attn_kernel_size": 1,
+  "model_dim": 152,
+  "dataset": "METR-LA",
+  "graph_mode": "learned",
+  "lambda_hybrid": 0.5,
+  "sparsity_k": 16,
+  "propagation_mode": "chebyshev",
+  "temporal_mode": "tcn",
+  "mamba_d_state": 16,
+  "mamba_d_conv": 4,
+  "mamba_expand": 2,
+  "tcn_num_layers": 3,
+  "tcn_kernel_size": 3,
+  "tcn_dilation_base": 2,
+  "tcn_dropout": 0.1,
+  "use_zero_init": false
+}

hub_metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "METR-LA",
+  "metrics": {},
+  "framework": "PyTorch",
+  "hf_repo_prefix": "STGFORMER_FINAL",
+  "implementation": "internal"
+}

metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "METR-LA",
+  "upload_date": "2025-12-11T17:07:45.721639",
+  "metrics": {},
+  "framework": "PyTorch",
+  "model_type": "STGFORMER_FINAL"
+}

model.safetensors

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

scaler.json

@@ -0,0 +1,4 @@
+{
+  "mean": 58.58370590209961,
+  "std": 12.823284149169922
+}