/DDSTGCN

Dual Dynamic Spatial-Temporal Graph Convolution Network for Traffic Prediction

Primary LanguagePython

DDSTGCN

Y. Sun, X. Jiang, Y. Hu, F. Duan, K. Guo, B. Wang, J. Gao, B. Yin, "Dual Dynamic Spatial-Temporal Graph Convolution Network for Traffic Prediction," in IEEE Transactions on Intelligent Transportation Systems, 2022, doi: 10.1109/TITS.2022.3208943.

『The code is still being sorted out, and a more complete version will be released later...』

Requirements

  • python 3
  • numpy
  • torch

Train Commands

python train.py

Dataset

Dataset URL: https://drive.google.com/drive/folders/1uoY8ROQU73BqWyl566ZNdRBOOTM4T2DS?usp=sharing