Shortcuts

SpamBase

class torch_uncertainty.datasets.classification.uci.SpamBase(root, transform=None, target_transform=None, binary=True, download=False, train=True, test_split=0.2, split_seed=21893027)[source]

The SpamBase UCI classification dataset.

Parameters:
  • root (str) – Root directory of the datasets.

  • train (bool, optional) – If True, creates dataset from training set, otherwise creates from test set.

  • transform (callable, optional) – A function/transform that takes in a numpy array and returns a transformed version.

  • target_transform (callable, optional) – A function/transform that takes in the target and transforms it.

  • download (bool, optional) – If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.

  • binary (bool, optional) – Whether to use binary classification. Defaults to True.

Note - License:

The licenses of the datasets may differ from TorchUncertainty’s license. Check before use.