Project: iterative-stratification

Package that provides scikit-learn compatible cross validators with stratification for multilabel data

Project Details

Latest version
0.1.7
Home Page
https://github.com/trent-b/iterative-stratification
PyPI Page
https://pypi.org/project/iterative-stratification/

Project Popularity

PageRank
0.002684097210930258
Number of downloads
38471

This iterative-stratification project offers implementations of MultilabelStratifiedKFold, MultilabelRepeatedStratifiedKFold, and MultilabelStratifiedShuffleSplit with a base algorithm for stratifying multilabel data described in the following paper: Sechidis K., Tsoumakas G., Vlahavas I. (2011) On the Stratification of Multi-Label Data. In: Gunopulos D., Hofmann T., Malerba D., Vazirgiannis M. (eds) Machine Learning and Knowledge Discovery in Databases. ECML PKDD 2011. Lecture Notes in Computer Science, vol 6913. Springer, Berlin, Heidelberg.