ltbio.ml.datasets.EventDetectionDataset

Overview

Classes

EventDetectionDataset

An abstract class representing a dataset of Biosignals.

Contents

class ltbio.ml.datasets.EventDetectionDataset.EventDetectionDataset(*objects: ltbio.biosignals.modalities.Biosignal.Biosignal, event_names: str | tuple[str], paddings: tuple[timedelta | int | None] = (None, None), ignore_margins: tuple[timedelta | int | None] = (None, None), name: str = None)

Bases: ltbio.ml.datasets.BiosignalDataset.BiosignalDataset

An abstract class representing a dataset of Biosignals. All subclasses should store the ordered list of objects and targets, respectively, in __objects and __targets. Also, subclasses have to overwrite __getitem__, supporting fetching an example for a given key. An example is a pair (object, target).

property class_weights: tuple[float, float]

balance_with_augmentation(*techniques: ltbio.ml.datasets.augmentation.DatasetAugmentationTechnique)

draw_timeline(precision: float)