Upload folder using huggingface_hub

README.md

@@ -0,0 +1,57 @@
+---
+tags:
+- traffic-forecasting
+- time-series
+- graph-neural-network
+- stgformer_pretrained
+datasets:
+- largest-gla
+---
+
+# Spatial-Temporal Graph Transformer (Pretrained) - LARGEST-GLA
+
+Spatial-Temporal Graph Transformer (Pretrained) (STGFORMER_PRETRAINED) trained on LARGEST-GLA dataset for traffic speed forecasting.
+
+## Model Description
+
+STGFormer pretrained checkpoint for LARGEST-GLA. This checkpoint contains pretrained model weights and imputation head from masked node pretraining. Use with load_from config option.
+
+
+
+## Dataset
+
+**LARGEST-GLA**: Traffic speed data from highway sensors.
+
+## Usage
+
+```python
+from utils.stgformer import load_from_hub
+
+# Load model from Hub
+model, scaler = load_from_hub("LARGEST-GLA", hf_repo_prefix="STGFORMER_PRETRAINED")
+
+# Get predictions
+from utils.stgformer import get_predictions
+predictions = get_predictions(model, scaler, test_dataset)
+```
+
+## Training
+
+Model was trained using the STGFORMER_PRETRAINED implementation with default hyperparameters.
+
+## Citation
+
+If you use this model, please cite the original STGFORMER_PRETRAINED 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.

hub_metadata.json

@@ -0,0 +1,6 @@
+{
+  "dataset": "LARGEST-GLA",
+  "checkpoint_type": "pretrained",
+  "framework": "PyTorch",
+  "hf_repo_prefix": "emelle/STGFormer-pretrain"
+}

imputation_head.safetensors

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

metadata.json

@@ -0,0 +1,7 @@
+{
+  "dataset": "LARGEST-GLA",
+  "upload_date": "2025-12-11T07:31:38.058165",
+  "metrics": {},
+  "framework": "PyTorch",
+  "model_type": "STGFORMER_PRETRAINED"
+}

model.safetensors

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

pretrain_config.json

@@ -0,0 +1,16 @@
+{
+  "dataset_name": "LARGEST-GLA",
+  "pretrain_config": {
+    "stage1_epochs": 10,
+    "stage1_mask_ratio": 0.15,
+    "stage2_epochs": 10,
+    "stage2_mask_ratio": 0.1,
+    "use_normalized_data": true,
+    "pretrain_data_fraction": 0.1,
+    "pretrain_batch_size": 8,
+    "learning_rate": 0.001,
+    "save_to": "emelle/STGFormer-pretrain"
+  },
+  "model_dim": 128,
+  "num_nodes": 3834
+}