Skip to content
Snippets Groups Projects
Commit 7ce4e3a8 authored by Moritz Huebner's avatar Moritz Huebner
Browse files

Merge branch 'store_prior_values' into 'master'

add storing prior probability of samples

See merge request Monash/tupak!159
parents 69311d09 df0da84e
No related branches found
No related tags found
1 merge request!159add storing prior probability of samples
Pipeline #30822 passed
......@@ -10,6 +10,7 @@ from collections import OrderedDict
from tupak.core import utils
from tupak.core.utils import logger
from tupak.core.prior import DeltaFunction
def result_file_name(outdir, label):
......@@ -435,6 +436,24 @@ class Result(dict):
data_frame = conversion_function(data_frame, likelihood, priors)
self.posterior = data_frame
def calculate_prior_values(self, priors):
"""
Evaluate prior probability for each parameter for each sample.
Parameters
----------
priors: dict, PriorSet
Prior distributions
"""
self.prior_values = pd.DataFrame()
for key in priors:
if key in self.posterior.keys():
if isinstance(priors[key], DeltaFunction):
continue
else:
self.prior_values[key]\
= priors[key].prob(self.posterior[key].values)
def construct_cbc_derived_parameters(self):
""" Construct widely used derived parameters of CBCs """
self.posterior['mass_chirp'] = (
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment