generate_artificial_indexes#
- mlquantify.utils.general.generate_artificial_indexes(y, prevalence: list, sample_size: int, classes: list)[source]#
Generate 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.