ml4gw.transforms.snr_rescaler

Classes

SnrRescaler(num_channels, sample_rate, ...)

class ml4gw.transforms.snr_rescaler.SnrRescaler(num_channels, sample_rate, waveform_duration, highpass=None, lowpass=None, dtype=torch.float32)

Bases: FittableSpectralTransform

Parameters:

  • num_channels (int)

  • sample_rate (float)

  • waveform_duration (float)

  • highpass (float | None)

  • lowpass (float | None)

  • dtype (dtype)

fit(*background, fftlength=None, overlap=None)
Parameters:

  • background (Float[Tensor, 'channel time'])

  • fftlength (float | None)

  • overlap (float | None)

forward(responses, target_snrs=None)
Parameters:

  • responses (Float[Tensor, 'batch num_ifos time'])

  • target_snrs (Float[Tensor, 'batch'] | None)