Skip to content

tsinghua-fib-lab/Traffic-Benchmark

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Traffic Prediction Benchmark

Python 3.6 PyTorch 1.7 License CC BY-NC-SA

This is the origin Pytorch implementation of DGCRN together with baselines in the following paper:



Figure 1. The architecture of DGCRN.

Model Classification

We classify the related works based on four parts: spatial topology construction, spatial dependency modeling, temporal dependency modeling and external features, as shown in the figure below, where the baselines in the benchmark are In bold type.



Figure 2. The classification of related works .

Requirements

  • Python 3.6
  • numpy == 1.19.4
  • pandas == 1.1.1
  • torch >= 1.1
  • mxnet == 1.4.1
  • tensorflow >= 2.4.0

Dependencies can be installed using the following command:

pip install -r requirements.txt

Data

  • The description of METR-LA dataset and PEMS_BAY dataset please refers to the repository of DCRNN.

  • The NE-BJ dataset used in the paper can be downloaded, which will be available soon.



Figure 3. Road segment distribution of the NE-BJ dataset.



Figure 4. Comparisons among datasets.

Usage

Commands for training model:

python train_benchmark.py --model 'model_name' --data 'data_name' >> log.txt 

More parameter information can be found in train_benchmark.py or the file in the directory of corrsponding model. You can refer to these parameters for experiments, and you can also adjust the parameters to obtain better results.

Results



Figure 5. Results of benchmark.

Citation

If you find this repository useful in your research, please consider citing the following paper:

@misc{li2021dynamic,
      title={Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solution}, 
      author={Fuxian Li and Jie Feng and Huan Yan and Guangyin Jin and Depeng Jin and Yong Li},
      year={2021},
      eprint={2104.14917},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

Moreover, the PyTorch version of baselines like GMAN and STSGCN is on the way~