CovidCTDataset#
- class CovidCTDataset(classes, path, **kwards)#
Bases:
DatasetWrapper
A wrapper for the _CovidCT2A_Dataset
- __init__(classes, path, **kwards)#
Methods
__init__
(classes, path, **kwards)get_approximation_set
()returns the approximation data set, to be used for range approximation
get_class_labels_dict
()Returns the class_name to index mapping
get_samples_per_class
(dataset)Returns the number of samples in each class or None (if the dataset is balanced - None is equivalent to a list with equal numbers) :param dataset: the dataset split :type dataset: Dataset
get_test_data
()Returns the test dataset
get_train_data
()Returns the train dataset
get_train_pipe_ffcv
(args)Returns the Dictionary defining for each field the sequence of ffcv Decoders and transforms to apply.
get_val_data
()Returns the validation dataset
is_imbalanced
()Returns True if the data is imbalanced