InputConfig

class torch_ecg.components.InputConfig(*args: CFG | dict, input_type: Literal['waveform', 'fft', 'spectrogram'], n_channels: int, n_samples: int = -1, ensure_batch_dim: bool = True, **kwargs: dict)

Bases: CFG

A Class to store the configuration of the input.

Parameters:

• input_type ({"waveform", "fft", "spectrogram"}) – Type of the input.

• n_channels (int) – Number of channels of the input.

• n_samples (int) – Number of samples of the input.

• ensure_batch_dim (bool) –

  Whether to ensure the transformed input has a batch dimension.

  Added in version 0.0.26.

Examples

input_config = InputConfig(
    input_type="waveform",
    n_channels=12,
    n_samples=5000,
)