Skip to content

Repository for the paper "Graph Convolutional Networks for Traffic Forecasting with Missing Values" in DMKD'22

License

Notifications You must be signed in to change notification settings

JingweiZuo/GCN-M

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GCN-M: Graph Convolutional Networks for Traffic Forecasting with missing values

This is the companion repository for our paper titled Graph Convolutional Networks for Traffic Forecasting with missing values published in Data Mining and Knowledge Discovery and also available on ArXiv.

Requirements:

  • matplotlib == 3.2.1
  • numpy == 1.19.2
  • pandas == 0.25.1
  • scikit_learn == 0.21.2
  • torch == 1.6.0
  • tensorwatch == 0.9.1

Dependencies can be installed using the following command:

pip install -r requirements.txt

Data

Step1:

  • Download METR-LA and PEMS-BAY data from Google Drive or Baidu Yun links provided by DCRNN.
  • Put the downloaded data into the repository mentioned in "config/DATASET.conf"

Step2: Preprocess raw data

python data/generate_dated_data.py

Usage

python main.py --config CONFIG_FILE --itr NBR_ITERATION

Citation

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

@article{zuo2023graph,
  title     = {Graph convolutional networks for traffic forecasting with missing values},
  author    = {Zuo, Jingwei and Zeitouni, Karine and Taher, Yehia and Garcia-Rodriguez, Sandra},
  journal   = {Data Mining and Knowledge Discovery},
  volume    = {37},
  number    = {2},
  pages     = {913--947},
  year      = {2023},
  publisher = {Springer}
}

About

Repository for the paper "Graph Convolutional Networks for Traffic Forecasting with Missing Values" in DMKD'22

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages