Randomized stratified k-fold cross-validation

User 2568 | 3/15/2016, 5:44:33 AM

The dataset I'm working on is highly unbalanced in the target variable. I could not find the equivalent of the scikit-learn [crossvalidation.StratifiedKFold](http://scikit-learn.org/stable/modules/generated/sklearn.crossvalidation.StratifiedKFold.html#sklearn.cross_validation.StratifiedKFold)

I found kfold (and random_split) but these don't support stratification and shuffle is a separate function.

Comments

User 1190 | 3/16/2016, 6:07:12 AM

Thank you for your feedback. This is definitely on our roadmap. We will take your feature request.