Skip to content

Latest commit

 

History

History
17 lines (11 loc) · 417 Bytes

README.md

File metadata and controls

17 lines (11 loc) · 417 Bytes

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