incerto.data.filter_dataset_by_class

incerto.data.filter_dataset_by_class#

incerto.data.filter_dataset_by_class(dataset, classes, invert=False)[source]#

Filter dataset to only include (or exclude) specific classes.

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

  • classes (List[int]) – List of class indices to keep

  • invert (bool) – If True, exclude specified classes instead

Return type:

Subset

Returns:

Subset containing filtered samples