core.utils.torchutils.build_train_and_test_loaders

core.utils.torchutils.build_train_and_test_loaders(
    dataset,
    train_fraction,
    batch_size,
    num_workers,
)

Split the dataset into train and test sets, and build corresponding DataLoaders. The random split uses a fixed seed for reproducibility.

Parameters

Name	Type	Description	Default
dataset	torch.utils.data.Dataset		required
train_fraction	float	Fraction of dataset to use for training. The remainder is used for testing. Should lie between 0 and 1.	required
batch_size	int		required
num_workers	int		required

Name	Type	Description
	(train_loader, test_loader)