CovidCTDataset#

class CovidCTDataset(classes, path, **kwards)#

A wrapper for the _CovidCT2A_Dataset

__init__(classes, path, **kwards)#

Initializes the CovidCTDataset wrapper for the _CovidCT2A_Dataset. This class is registered under the name ‘COVID_CT’ in the DSFactory.

Parameters:

classes (int) – The number of classes in the dataset.
path (str) – The base directory where the dataset is stored.
**kwargs – Additional keyword arguments. Not used. (Maintained for consistency with other dataset constructors)

Methods

`__init__`(classes, path, **kwards)	Initializes the CovidCTDataset wrapper for the _CovidCT2A_Dataset.
`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