yamle.data.custom module

class yamle.data.custom.ECG5000DataModule(*args, **kwargs)

Bases: BaseDataModule

This is the base class for the EC5000 dataset.

The link to download the dataset is: https://www.timeseriesclassification.com/description.php?Dataset=ECG5000

inputs_dim = (140, 1)

get_transform(name)

This is a helper function to get the transform by name.

Return type:

Callable[..., Any]

setup(augmentation=None)

Split the data into training, validation and test sets.

Additionally for apply augmentation to the test data. Insert the augmentation into the existing test transformation at the first position.

Parameters:

augmentation (str) – Name of the augmentation to apply to the training data. Default: None.

Return type:

None

prepare_data()

This function is used to download the dataset.

Return type:

Tuple[DataFrame, DataFrame]