Описание
Это бенчмарк скрипт для хакатона от Раййфайзенбанка по оценке коммерческой недвижимости Бенчмарк состоит из:
- pyproject.toml - конфигурационный файл для менеджера пакетов poetry (https://python-poetry.org/) - в интернете есть много статей, посвященных ему (например https://habr.com/ru/post/455335/ и https://khashtamov.com/ru/python-poetry-dependency-management/)
- requirements.txt - стандартный requirements для pip
- train.py - скрипт, который обучает модель и сохраняет ее
- predict.py - скрипт, который делает предсказание на отложенной тестовой выборке
Запуск
Установка
pip install poetry
poetry install
Запустить обучение
poetry run python3 train.py --train_data <path_to_train_data> --model_path <path_to_pickle_ml_model>
Запустить предикт
poetry run python3 predict.py --model_path <path_to_pickled_model> --test_data <path_to_test_data> --output <path_to_output_csv_file>