incerto.data.get_class_balanced_subset

incerto.data.get_class_balanced_subset#

incerto.data.get_class_balanced_subset(dataset, samples_per_class, seed=42)[source]#

Create a class-balanced subset with equal samples per class.

Parameters:
  • dataset (Dataset) – Dataset with ‘targets’ or ‘labels’ attribute

  • samples_per_class (int) – Number of samples per class

  • seed (int) – Random seed

Return type:

Subset

Returns:

Balanced subset