strata_from_labels convenience function for cross-validation, doc bugfix