From 3def4258c07bf5df6bb4d6e1f633d67e3029284f Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Thu, 27 Sep 2018 12:46:23 +1000
Subject: [PATCH] store likelihood in dynesty nested samples
---
tupak/core/sampler/dynesty.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tupak/core/sampler/dynesty.py b/tupak/core/sampler/dynesty.py
index 61fa1a54..dd84e473 100644
--- a/tupak/core/sampler/dynesty.py
+++ b/tupak/core/sampler/dynesty.py
@@ -131,11 +131,11 @@ class Dynesty(Sampler):
# self.result.sampler_output = out
weights = np.exp(out['logwt'] - out['logz'][-1])
- self.result.samples = dynesty.utils.resample_equal(
- out.samples, weights)
+ self.result.samples = dynesty.utils.resample_equal(out.samples, weights)
self.result.nested_samples = DataFrame(
out.samples, columns=self.search_parameter_keys)
self.result.nested_samples['weights'] = weights
+ self.result.nested_samples['log_likelihood'] = out.logl
idxs = [np.unique(np.where(self.result.samples[ii] == out.samples)[0])
for ii in range(len(out.logl))]
self.result.log_likelihood_evaluations = out.logl[idxs]
--
GitLab