Стратифікований K-fold

Щойно ви помітили, що через випадкові поділи розподіл цільової змінної у фолдах суттєво різниться. Для цієї конкретної змагання це не критично, але у задачах класифікації з сильно незбалансованою цільовою змінною це може стати проблемою.

Щоб це усунути, реалізуймо стратіфіковану стратегію K-fold зі стратифікацією за цільовою змінною. Датафрейм train уже доступний у вашому робочому середовищі.