get_indexes_with_prevalence

mlquantify.utils.get_indexes_with_prevalence(y, prevalence: list, sample_size: int, random_state: int | None = None)

Get indexes for a stratified sample based on the prevalence of each class.

Parameters:

ynp.ndarray

Array of class labels.

prevalencelist

List of prevalences for each class.

sample_sizeint

Number of samples to generate.

classeslist

List of unique classes.

Returns:

list

List of indexes for the stratified sample.