gw.inference.gw_samplers.GWSampler

gw.inference.gw_samplers.GWSampler(**kwargs)

Sampler for gravitational-wave inference using neural posterior estimation. Augments the base class by defining transform_pre and transform_post to prepare data for the inference network.

transform_pre : * Decimates data (if necessary and using MultibandedFrequencyDomain). * Whitens strain. * Repackages strain data and the inverse ASDs (suitably scaled) into a torch tensor.

transform_post : * Extract the desired inference parameters from the network output ( array-like), de-standardize them, and repackage as a dict.

Also mixes in GW functionality for building the domain and correcting the reference time.

Allows for conditional and unconditional models, and draws samples from the model based on (optional) context data.

This is intended for use either as a standalone sampler, or as a sampler producing initial sample points for a GNPE sampler.