vision_unlearning.datasets.imagenette

Classes

UnlearnDatasetImagenette

Wrapper around huggingface datasets

Module Contents

class vision_unlearning.datasets.imagenette.UnlearnDatasetImagenette(/, **data: Any)[source]

Bases: vision_unlearning.datasets.base.UnlearnDataset

Wrapper around huggingface datasets Organize the forget-retain splits

download_path: str

class_mapping: ClassVar[dict]

_download_imagenette(temp_download_path: str) → None[source]: Download Imagenette from FastAI’s S3 bucket to a temporary folder. Skip download if files already exist.

_load() → None[source]: Load the dataset from disk or download it. Side effects: updates the properties _dataset_splits, _classes, _n_classes

make_prompt_for_label(label: int) → str[source]