OctopusLi / NeurIPS2022-traffic4cast

Traffic4cast2022

Data Preparation

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.ipynb

Usage

cd model
run cluster.ipynb
python GNN_model_train.py
python GNN_model_test.py
python submission_cc.py
python submission_eta.py

Acknowledgements

This repository is based on NeurIPS2022-Traffic4cast from the Institute of Advanced Research in Artificial Intelligence (IARAI).