oasis.stratification.auto_stratify

oasis.stratification.auto_stratify(scores, **kwargs)

Generate Strata instance automatically

Parameters

scoresarray-like, shape=(n_items,)

ordered array of scores which quantify the classifier confidence for the items in the pool. High scores indicate a high confidence that the true label is a “1” (and vice versa for label “0”).

**kwargs :

optional keyword arguments. May include ‘stratification_method’, ‘stratification_n_strata’, ‘stratification_n_bins’.

Returns

Strata instance