Home

Awesome

Traffic4cast2022

Data Preparation

After cloning this repo, download and extract data, cache and output to the root of the cloned repo.

Usage

To create a submission for the Core Challenge, run

python inference_first_stage.py --city london
python inference_first_stage.py --city madrid
python inference_first_stage.py --city melbourne
python preprocess_core_xgb.py --city london
python preprocess_core_xgb.py --city madrid
python preprocess_core_xgb.py --city melbourne
python inference_core_xgb.py --city london
python inference_core_xgb.py --city madrid
python inference_core_xgb.py --city melbourne
python write_submission_core.py

This creates a submission file named cc_submission.zip under the output folder.

To create a submission for the Extended Challenge, run

python inference_first_stage.py --city london
python inference_first_stage.py --city madrid
python inference_first_stage.py --city melbourne
python preprocess_extended_lgb.py --city london
python preprocess_extended_lgb.py --city madrid
python preprocess_extended_lgb.py --city melbourne
python preprocess_extended_lgb_v2.py --city london
python preprocess_extended_lgb_v2.py --city madrid
python preprocess_extended_lgb_v2.py --city melbourne
python inference_extended_lgb.py --city london
python inference_extended_lgb.py --city madrid
python inference_extended_lgb.py --city melbourne
python write_submission_extended.py

This creates a submission file named eta_submission.zip under the output folder.