gw.training.train_builders.set_train_transforms

gw.training.train_builders.set_train_transforms(
    wfd,
    data_settings,
    asd_dataset_path,
    omit_transforms=None,
)

Set the transform attribute of a waveform dataset based on a settings dictionary. The transform takes waveform polarizations, samples random extrinsic parameters, projects to detectors, adds noise, and formats the data for input to the neural network. It also implements optional GNPE transformations.

Note that the WaveformDataset is modified in-place, so this function returns nothing.

Parameters

Name	Type	Description	Default
wfd	WaveformDataset		required
data_settings	dict		required
asd_dataset_path	str	Path corresponding to the ASD dataset used to generate noise.	required
omit_transforms		List of sub-transforms to omit from the full composition.	None