nessai.gw.proposal#

Specific proposal methods for sampling gravitational-wave models.

Classes#

GWFlowProposal

Wrapper for FlowProposal that has defaults for CBC-PE

AugmentedGWFlowProposal

Augmented version of GWFlowProposal.

Module Contents#

class nessai.gw.proposal.GWFlowProposal(model, poolsize=None, latent_prior='truncated_gaussian', latent_temperature=None, constant_volume_mode=True, volume_fraction=0.95, fuzz=1.0, fixed_radius=False, drawsize=None, truncate_log_q=False, expansion_fraction=4.0, min_radius=False, max_radius=50.0, compute_radius_with_all=False, enforce_likelihood_threshold=False, **kwargs)#

Bases: nessai.proposal.FlowProposal

Wrapper for FlowProposal that has defaults for CBC-PE

aliases#

Dictionary of aliases used to determine the default reparameterisations for common gravitational-wave parameters.

use_default_reparameterisations = True#

GW specific reparameterisations will be used by default. This is different to the parent class where they are disabled by default.

get_reparameterisation(reparameterisation)#

Function to get reparameterisations that checks GW defaults and aliases

add_default_reparameterisations()#

Add default reparameterisations for parameters that have not been specified.

class nessai.gw.proposal.AugmentedGWFlowProposal(model, augment_dims=1, generate_augment='gaussian', marginalise_augment=False, n_marg=50, **kwargs)#

Bases: nessai.proposal.AugmentedFlowProposal, GWFlowProposal

Augmented version of GWFlowProposal.

See AugmentedFlowProposal and GWFlowPropsosal