Skip to content

Latest commit

 

History

History
24 lines (20 loc) · 957 Bytes

README.md

File metadata and controls

24 lines (20 loc) · 957 Bytes
Raw

GSTNet (This ReadMe file is updating)

This is the source code of GSTNet.

The PeMS-BAY Dataset is provided by DCRNN, and preprocessed here: https://drive.google.com/drive/folders/10L1ZkwHNIxaoOYRw3Kw8bzhJ0obTeCBl?usp=sharing

Please download the dataset and put it in the "DCRNN_Provide" folder.

To re-produce the results on PeMS-BAY, just run the "Main.py" file.

If this source code is helpful to you, please cite the following paper.

Update Notice：Recently I was told that the there lacks necessary modules in the "DCRNN_Provide" folder. This bug will be fixed as soon as possible.

@inproceedings{Fang_GSTNet,
  author    = {Shen Fang and
               Qi Zhang and
               Gaofeng Meng and
               Shiming Xiang and
               Chunhong Pan},
  title     = {{GSTNet}: Global Spatial-Temporal Network for Traffic Flow Prediction},
  booktitle = {Proc. 28th IJCAI},
  pages     = {2286--2293},
  year      = {2019},
}