Views
No views yet
1from utils.stgformer import load_from_hub
2
3# Load model from Hub
4model, scaler = load_from_hub("METR-LA")
5
6# Get predictions
7import numpy as np
8x = np.random.randn(10, 12, 207, 2) # (batch, seq_len, nodes, [value, tod])
9predictions = model.predict(x)1@article{stgformer,
2 title={STGformer: Spatio-Temporal Graph Transformer for Traffic Forecasting},
3 author={Author names},
4 journal={Conference/Journal},
5 year={Year}
6}