BalancedBatchSampler

class humancompatible.train.fairness.utils.BalancedBatchSampler(group_onehot: Iterable[Iterable[int]] | None | Tensor = None, group_indices: Iterable[Iterable[int]] | None = None, batch_size: int = 1, drop_last: bool = True, extend_groups: Iterable[int] | None = None, generator: Generator | None = None)