Запуск

Чтобы воспроизвести наш результат, надо воспроизвести следующие шаги:

Убедиться, что у вас стоит python3.6 или выше

Установить зависимости:

pip install -r requirements.txt 

Расположить данные в правильной папке: скопировать файлы train.csv и test.csv в подпапку data. В итоге должна получиться следующая структура:

raif_gg_hack
    data
        train.csv
        test.csv
    raif_hack
        ...
    train.py
    predict.py
    validation_split.py
    ...

Выделить валидационный сет из обучающей выборки:

python3 validation_split.py

Запустить обучение:

python3 train.py --model_path model.pkl --val

Запустить инференс:

python3 predict.py --model_path model.pkl --test_data data/test.csv --output final_submission.csv

Предсказания содержатся в файле final_submission.csv