/ST-GWCN

A traffic flow prediction model based on graph wavelet convolution network and gated temporal convolution network

Primary LanguagePython

Related works

  1. TCN: An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling
  2. GLU and GTU: Language Modeling with Gated Convolutional Networks
  3. ChebyNet: Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering

Related code

  1. TCN: https://github.com/locuslab/TCN
  2. ChebyNet: https://github.com/mdeff/cnn_graph

Dataset

Source

  1. METR-LA: DCRNN author's Google Drive
  2. PEMS-BAY: DCRNN author's Google Drive
  3. PeMSD7(M): STGCN author's GitHub repository

After prepare dataset, run: python main_road_traffic.py