CollectionDataset

class torchhd.datasets.CollectionDataset(root: str, transform: Callable | None = None, target_transform: Callable | None = None, download: bool = False)

用于加载在Do we Need Hundreds of Classifiers to Solve Real World Classification Problems?中使用的数据集的通用类。

Parameters:

• root (string) – 包含数据集文件的根目录。

• transform (callable, optional) – 一个函数/转换，它接收一个 torch.FloatTensor 并返回一个转换后的版本。

• target_transform (callable, optional) – 一个函数/转换，它接收目标并对其进行转换。

• 下载 (布尔型, 可选) – 如果为True，则从互联网下载数据集并将其放置在根目录中。如果数据集已经下载，则不会再次下载。

Special-members:

__getitem__(index: int) → Tuple[FloatTensor, LongTensor]

Parameters:

index (int) – 索引

Returns:

(sample, target) 其中 target 是目标类的索引

Return type:

元组[torch.FloatTensor, torch.LongTensor]