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