/STGAT

Implementation of Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting(for IJCAI 2020)

Primary LanguagePythonMIT LicenseMIT

STGAT: Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting

This is the implementation of Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting(for IJCAI 2020) More details will be added ...

train.py util.py run_demo.py model_stgat.py draw.py experiment

data # file

baselines train_base.py run_demo_base.py gwnet.py rnn.py stgcn.py experiment_base #file