Upload STGFORMER_TCN model trained on METR-LA

README.md

@@ -0,0 +1,57 @@
+---
+tags:
+- traffic-forecasting
+- time-series
+- graph-neural-network
+- stgformer_tcn
+datasets:
+- metr-la
+---
+
+# Spatial-Temporal Graph Transformer (Tcn) - METR-LA
+
+Spatial-Temporal Graph Transformer (Tcn) (STGFORMER_TCN) trained on METR-LA dataset for traffic speed forecasting.
+
+## Model Description
+
+STGFormer with TCN temporal processing (causal dilated convolutions)
+
+
+
+## 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_TCN")
+
+# Get predictions
+from utils.stgformer import get_predictions
+predictions = get_predictions(model, scaler, test_dataset)
+```
+
+## Training
+
+Model was trained using the STGFORMER_TCN implementation with default hyperparameters.
+
+## Citation
+
+If you use this model, please cite the original STGFORMER_TCN 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,32 @@
+{
+  "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": 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": "METR-LA",
+  "graph_mode": "learned",
+  "lambda_hybrid": 0.5,
+  "sparsity_k": null,
+  "propagation_mode": "power",
+  "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": true
+}

hub_metadata.json

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

metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "METR-LA",
+  "upload_date": "2025-12-07T17:39:48.517559",
+  "metrics": {},
+  "framework": "PyTorch",
+  "model_type": "STGFORMER_TCN"
+}

model.safetensors

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