Stratyfikowany K-fold

Jak właśnie zauważasz, rozkład zmiennej docelowej różni się dość znacznie między foldami ze względu na losowe podziały. W przypadku tego konkretnego konkursu nie jest to kluczowy problem, jednak może stanowić poważne wyzwanie w konkursach klasyfikacyjnych z mocno niezrównoważoną zmienną docelową.

Aby sobie z tym poradzić, zaimplementuj strategię stratyfikowanego K-fold z podziałem według zmiennej docelowej. DataFrame train jest już dostępny w twoim środowisku pracy.