Awesome

CoGNN

Dataset

The dataset can be downloaded from here. The downloaded datasets are suggested to be stored in the ./data/.

Training

To train CoGNN, you can run the scripts train.py such as:

python train.py --data 'data/train_val_test_data/pickup