vision_unlearning.datasets.cifar

Classes

UnlearnDatasetCifar

Wrapper around huggingface datasets

Module Contents

class vision_unlearning.datasets.cifar.UnlearnDatasetCifar(/, **data: Any)

Bases: vision_unlearning.datasets.base.UnlearnDataset

Wrapper around huggingface datasets Organize the forget-retain splits

download_path: str

_load() → None

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