/MRA-BGCN

PyTorch implementation of Multi-Range Attentive Bicomponent Graph Convolutional Network

Primary LanguagePython

MRA-BGCN

PyTorch implementation of Multi-Range Attentive Bicomponent Graph Convolutional Network (MRA-BGCN) in the following AAAI2020 paper "Multi-Range Attentive Bicomponent Graph Convolutional Network for Traffic Forecasting".

The dataset only uses PEMS08. You are welcome to point out the code error in the comment area.

Requirements

  • python 3
  • numpy
  • torch

Train Commands

python train.py