SplitDataset¶

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

SplitDataset is used to create Dataset Subset using indices.

Parameters

dataset¶ (Any) – A dataset to be splitted
indices¶ (Optional[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)