Need help with STGCN-PyTorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

186 Stars 53 Forks MIT License 3 Commits 7 Opened issues


🚗 Implementation of spatio-temporal graph convolutional network with PyTorch

Services available


Need anything else?

Contributors list

No Data


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.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.