Как справиться с дисбалансом класса с sklearn GroupKfold?

Question

Sklearn имеет StratifiedKFold, но не имеет стратифицированного GroupKFold. У кого-нибудь есть хороший рабочий процесс для дисбаланса классов в сгруппированных данных?

Answer 1

Sklearn имеет GroupKFlod http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.GroupKFold.html. Это может помочь вам создать свой собственный класс.