Skip to content

ecg_synthetic

Classes

EcgSyntheticDataloader 

EcgSyntheticDataloader(ds: EcgSyntheticDataset, **kwargs)

Dataloader for ECG synthetic dataset

Source code in heartkit/tasks/segmentation/dataloaders/ecg_synthetic.py 
26
27
28
29
30
31
32
33
def __init__(self, ds: EcgSyntheticDataset, **kwargs):
    """Dataloader for ECG synthetic dataset"""
    super().__init__(ds=ds, **kwargs)
    # Update label map
    if self.label_map:
        self.label_map = {
            k: self.label_map[v] for (k, v) in EcgSyntheticSegmentationMap.items() if v in self.label_map
        }

Functions