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