j1o2h3n / STSGCN

PyTorch implementation of Spatial-Temporal Synchronous Graph Convolutional Networks

STSGCN

PyTorch implementation of Spatial-Temporal Synchronous Graph Convolutional Networks (STSGCN) in the following AAAI2020 paper "Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting".

『Code v1.0 version completed.....』

The completion time is relatively short, and the dataset only uses PEMS08. You are welcome to point out the code error in the comment area.

You can refer to the mxnet code provided by the author of the original paper.

Requirements

• python 3
• numpy
• torch

Train Commands

python train.py