BankMarketingDataModule#
- class torch_uncertainty.datamodules.BankMarketingDataModule(root, batch_size, eval_batch_size=None, val_split=0.0, test_split=0.2, num_workers=1, pin_memory=True, persistent_workers=True, binary=True)[source]#
The Bank Marketing UCI classification datamodule.
- Parameters:
root (str | Path) – Root directory of the datasets.
batch_size (int) – The batch size for training and testing.
eval_batch_size (int | None) – Number of samples per batch during evaluation (val and test). Set to
batch_size
ifNone
. Defaults toNone
.val_split (float, optional) – Share of validation samples among the non-test samples. Defaults to
0
.test_split (float, optional) – Share of test samples. Defaults to
0.2
.num_workers (int, optional) – How many subprocesses to use for data loading. Defaults to
1
.pin_memory (bool, optional) – Whether to pin memory in the GPU. Defaults to
True
.persistent_workers (bool, optional) – Whether to use persistent workers. Defaults to
True
.binary (bool, optional) – Whether to use binary classification. Defaults to
True
.