daozhong / STGCN-PyTorch

🚗 Implementation of spatio-temporal graph convolutional network with PyTorch

STGCN-PyTorch

PyTorch implementation of the spatio-temporal graph convolutional network proposed in Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting by Bing Yu, Haoteng Yin, Zhanxing Zhu. An example for traffic forecasting is included in this repository.

Alternative implementations using other deep learning frameworks can be found here:

• TensorFlow by PKUAI26
• MXNet by Davidham3

Requirements

• PyTorch
• NumPy
• Matplotlib

Example Dataset

The repository provides a usage example on the METR-LA dataset (original version to be found here). Unfortunately, there is no publicly available standard version of the PeMSD dataset used in the paper.