After downloading the [data] (https://github.com/iarai/NeurIPS2022-traffic4cast)) and checkpoints, run
prepare_training_data_cc.py -d <data folder with unzipped downloads>
prepare_training_data_eta.py -d <data folder with unzipped downloads>
prepare_training_check_labels.py -d <data folder with unzipped downloads>
cd data
run data_preprocess.ipynbcd model
run cluster.ipynb
python GNN_model_train.py
python GNN_model_test.py
python submission_cc.py
python submission_eta.pyThis repository is based on NeurIPS2022-Traffic4cast from the Institute of Advanced Research in Artificial Intelligence (IARAI).