- 🔝 As of AAAI 2022, our paper, "Graph Neural Controlled Differential Equations for Traffic Forecasting", was ranked 14th among the most influential papers.
This is the repository of our accepted AAAI 2022 paper "Graph Neural Controlled Differential Equations for Traffic Forecasting". Paper is available on arxiv.
If you find this code useful, you may cite us as:
@inproceedings{choi2022STGNCDE,
title={Graph Neural Controlled Differential Equations for Traffic Forecasting},
author={Jeongwhan Choi AND Hwangyong Choi AND Jeehyun Hwang AND Noseong Park},
booktitle={AAAI},
year={2022}
}
Install python environment
$ conda env create -f environment.yml
- Run the shell file (at the root of the project)
$ bash run.sh
- Run the python file (at the
model
folder)
$ cd model
$ python Run_cde.py --dataset='PEMSD4' --model='GCDE' --model_type='type1' --embed_dim=10 --hid_dim=64 --hid_hid_dim=64 --num_layers=2 --lr_init=0.001 --weight_decay=1e-3 --epochs=200 --tensorboard --comment="" --device=0