Graph Neural Controlled Differential Equations for Traffic Forecasting
Setup Python environment for STG-NCDE
Install python environment
$ conda env create -f environment.yml
Reproducibility
Usage
In terminal
- 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