Views
No views yet
1from utils.stgformer import load_from_hub
2
3# Load model from Hub
4model, scaler = load_from_hub("METR-LA", hf_repo_prefix="STGFORMER_PRETRAIN_STAGE2ONLY_IMPUTE")
5
6# Get predictions
7from utils.stgformer import get_predictions
8predictions = get_predictions(model, scaler, test_dataset)1@inproceedings{lan2022stgformer,
2 title={STGformer: Spatial-Temporal Graph Transformer for Traffic Forecasting},
3 author={Lan, Shengnan and Ma, Yong and Huang, Weijia and Wang, Wanwei and Yang, Hui and Li, Peng},
4 booktitle={IEEE Transactions on Neural Networks and Learning Systems},
5 year={2022}
6}