validation#

Classes

class datasets.utils.validation.ValidationDataset(data, targets, transform=None, target_transform=None)[source]#

Bases: Dataset

Functions

datasets.utils.validation.get_train_val(train, test_transform, dataset, val_perc=0.1)[source]#

Extract val_perc% of the training set as the validation set.

Parameters:

  • train (Dataset) – training dataset

  • test_transform (Module) – transformation of the test dataset

  • dataset (str) – dataset name

  • val_perc (float) – percentage of the training set to be extracted

Returns:

the training set and the validation set