Shortcuts

SplitDataset

class flash.core.data.splits.SplitDataset(dataset, indices, running_stage=None, use_duplicated_indices=False)[source]

SplitDataset is used to create Dataset Subset using indices.

Parameters
  • dataset (Any) – A dataset to be split

  • indices (List[int]) – List of indices to expose from the dataset

  • use_duplicated_indices (bool) – Whether to allow duplicated indices.

Example:

split_ds = SplitDataset(dataset, indices=[10, 14, 25])

split_ds = SplitDataset(dataset, indices=[10, 10, 10, 14, 25], use_duplicated_indices=True)