gwkokab.analysis.core.discrete_base¶

Classes¶

AnalysisBase is a class which contains all the common functionality among the

Functions¶

discrete_arg_parser(→ argparse.ArgumentParser)

Populate the command line argument parser with the arguments for the discrete

Module Contents¶

class gwkokab.analysis.core.discrete_base.DiscreteBase(likelihood_fn: collections.abc.Callable[Ellipsis, collections.abc.Callable[Ellipsis, jaxtyping.Array]], model: numpyro.distributions.Distribution | collections.abc.Callable[Ellipsis, numpyro.distributions.Distribution], where_fns: List[collections.abc.Callable[Ellipsis, jaxtyping.Array]] | None, data_loader: gwkokab.analysis.core.inference_io.DiscretePELoader, prior_filename: str, poisson_mean_filename: str, sampler_cfg, variance_cut_threshold: float | None, n_buckets: int | None, threshold: float, debug_nans: bool = False, profile_memory: bool = False, check_leaks: bool = False, analysis_name: str = '')¶

Bases: gwkokab.analysis.core.analysis_base.AnalysisBase

AnalysisBase is a class which contains all the common functionality among the different analyses.

It is not meant to be used directly, but rather to be subclassed by the specific analyses.

read_data() → Tuple[Tuple[jaxtyping.Array, Ellipsis], Tuple[jaxtyping.Array, Ellipsis], Tuple[jaxtyping.Array, Ellipsis]]¶

run() → None¶

data_loader¶

likelihood_fn¶

n_buckets¶

threshold¶

where_fns¶

gwkokab.analysis.core.discrete_base.discrete_arg_parser(parser: argparse.ArgumentParser) → argparse.ArgumentParser¶: Populate the command line argument parser with the arguments for the discrete analyses.