olgashipkova / churn-forecast Goto Github PK

В процессе работы над проектом была автоматизирована таблица результатов исследуемых моделей, это позволяет легко менять гиперпараметры, добавлять новые модели и автоматически получать данные не используя ручной ввод таблицы, что позволяет избежать возможных ошибок, связанных с ручным вводом данных и ускоряет процесс.
Также была добавлена модель градиентного бустинга, но это не позволило улучшить полученный результат.
В целях дальнейшего улучшения проекта рассматривается использование следующих моделей XGBClassifier, CatBoostClassifier, StackingClassifier, подбор гиперпараметров будет осуществляться с помощью GridSearchCV, как и для других моделей.