STGCN-PyTorch icon indicating copy to clipboard operation
STGCN-PyTorch copied to clipboard

verified publisher icon Aguin
→

Metadata

PyTorch implementation of STGCN

STGCN-PyTorch

Paper & Official Implementation

Yu, B., Yin, H., & Zhu, Z. (2017). Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875.

TF1 by VeritasYin

Alternative Implementations

MXNet by Davidham3

PyTorch by FelixOpolka

Results

Model PeMSD7(M) 15min
MAE MAPE(%) RMSE
STGCN(Cheb) 2.212 5.184 3.971

← Metadata

49
Stars
16
Forks
Watchers

Owner

verified publisher icon Aguin

Metadata

PyTorch implementation of STGCN

Back