I want to create a Stratified KFold with a flag variable which values are 0 and 1. Then I want to use
xgb.cv with this custom fold instance. If I create any KFold instance and use it as a fold in
xgb.cv parameters it works without any problem but when I want to use Stratifed KFold I got
IndexError: list index out of range error. When using KFold
folds = KFold() is enough. For Stratified one need to define split argument with y. So I’m trying with
folds = StratifiedKFold().split(X,y) and get index error. I can use lightgbm with that way without any problem. Is there any proper way to use it? Thanks in advance.