vision_unlearning.datasets.local

Classes

UnlearnDatasetLocal

Wrapper around huggingface datasets

Module Contents

class vision_unlearning.datasets.local.UnlearnDatasetLocal(/, **data: Any)

Bases: vision_unlearning.datasets.base.UnlearnDataset

Wrapper around huggingface datasets Organize the forget-retain splits

path: str

_load() → None

Load the dataset from disk or download it. Side effects: updates the properties _dataset_splits, _classes, _n_classes