image thumbnail

Stratified cross-validation for multi-label data sets

version 1.0.1 (2.07 KB) by Jan Motl
Greedily assign samples into folds based on their labels

30 Downloads

Updated 29 Oct 2019

View License

The algorithm partitions a binary multi-label Y into N folds. The returned vector assigns samples into the folds such way, that the same distribution of positive and negative samples of each label is approximately maintained across all folds.

The implementation can be used as a drop-in replacement for CVPARTITION.

Reference: On the Stratification of Multi-label Data by Sechidis, Konstantinos & Tsoumakas, Grigorios & Vlahavas, Ioannis.

Cite As

Jan Motl (2021). Stratified cross-validation for multi-label data sets (https://www.mathworks.com/matlabcentral/fileexchange/73003-stratified-cross-validation-for-multi-label-data-sets), MATLAB Central File Exchange. Retrieved .

MATLAB Release Compatibility
Created with R2018a
Compatible with any release
Platform Compatibility
Windows macOS Linux

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!