nessai.gw.proposal

Specific proposal methods for sampling gravitational-wave models.

Module Contents

Classes

GWFlowProposal	Wrapper for FlowProposal that has defaults for CBC-PE
AugmentedGWFlowProposal	Augmented version of GWFlowProposal.

Attributes

logger

nessai.gw.proposal.logger

class nessai.gw.proposal.GWFlowProposal(model, flow_config=None, output='./', poolsize=None, rescale_parameters=True, latent_prior='truncated_gaussian', constant_volume_mode=True, volume_fraction=0.95, fuzz=1.0, plot='min', fixed_radius=False, drawsize=None, check_acceptance=False, truncate=False, rescale_bounds=[- 1, 1], expansion_fraction=4.0, boundary_inversion=False, inversion_type='split', update_bounds=True, min_radius=False, max_radius=50.0, max_poolsize_scale=10, update_poolsize=True, save_training_data=False, compute_radius_with_all=False, draw_latent_kwargs=None, detect_edges=False, detect_edges_kwargs=None, reparameterisations=None, fallback_reparameterisation=None, use_default_reparameterisations=None, **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(self, reparameterisation)

Function to get reparameterisations that checks GW defaults and aliases

add_default_reparameterisations(self)

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.augmented.AugmentedFlowProposal, GWFlowProposal

Augmented version of GWFlowProposal.

See AugmentedFlowProposal and GWFlowPropsosal