dataloaders

Classes

DenoiseDataloader

DenoiseDataloader(**kwargs)

Generic Dataloader for denoising task.

Source code in heartkit/tasks/denoise/dataloaders.py

def __init__(self, **kwargs):
    """Generic Dataloader for denoising task."""
    super().__init__(**kwargs)

Functions

data_generator

data_generator(
    patient_ids: list[int], samples_per_patient: int | list[int], shuffle: bool = False
) -> Generator[npt.NDArray, None, None]

Generate data for given patient ids. Leveraging the signal_generator method from the dataset class to generate data.

Source code in heartkit/tasks/denoise/dataloaders.py

def data_generator(
    self,
    patient_ids: list[int],
    samples_per_patient: int | list[int],
    shuffle: bool = False,
) -> Generator[npt.NDArray, None, None]:
    """Generate data for given patient ids.
    Leveraging the signal_generator method from the dataset class to generate data.
    """
    gen = self.ds.signal_generator(
        patient_generator=helia.utils.uniform_id_generator(patient_ids, repeat=True, shuffle=shuffle),
        frame_size=self.frame_size,
        samples_per_patient=samples_per_patient,
        target_rate=self.sampling_rate,
    )
    for x in gen:
        x = np.nan_to_num(x, neginf=0, posinf=0).astype(np.float32)
        x = np.reshape(x, (-1, 1))
        yield x