From d6e765a2be76dca2df4ad0e453ca24941f063b95 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Thu, 27 Sep 2018 12:41:36 +1000
Subject: [PATCH] fix likelihoods stored in dynesty posterior
---
tupak/core/sampler/dynesty.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/tupak/core/sampler/dynesty.py b/tupak/core/sampler/dynesty.py
index ada5d3b21..61fa1a54d 100644
--- a/tupak/core/sampler/dynesty.py
+++ b/tupak/core/sampler/dynesty.py
@@ -133,12 +133,14 @@ class Dynesty(Sampler):
weights = np.exp(out['logwt'] - out['logz'][-1])
self.result.samples = dynesty.utils.resample_equal(
out.samples, weights)
- self.result.log_likelihood_evaluations = out.logl
- self.result.log_evidence = out.logz[-1]
- self.result.log_evidence_err = out.logzerr[-1]
self.result.nested_samples = DataFrame(
out.samples, columns=self.search_parameter_keys)
self.result.nested_samples['weights'] = weights
+ 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]
+ self.result.log_evidence = out.logz[-1]
+ self.result.log_evidence_err = out.logzerr[-1]
if self.plot:
self.generate_trace_plots(out)
--
GitLab