incerto.data.create_imbalanced_dataset

incerto.data.create_imbalanced_dataset#

incerto.data.create_imbalanced_dataset(dataset, imbalance_ratio=0.1, minority_classes=None, seed=42)[source]#

Create an imbalanced version of a dataset.

Parameters:
  • dataset (Dataset) – Original dataset

  • imbalance_ratio (float) – Ratio of minority to majority class size

  • minority_classes (Optional[List[int]]) – Classes to make minority (default: half of classes)

  • seed (int) – Random seed

Return type:

Subset

Returns:

Imbalanced subset