Subset

Contents

Subset#

class stable_ssl.data.Subset(dataset: Dataset, indices: Sequence[int])[source]#

Bases: Dataset

Subset of a dataset at specified indices.

Parameters:
  • dataset (Dataset) – The whole Dataset

  • indices (sequence) – Indices in the whole set selected for subset