From c0d1c423741658ddf87516e4ac41993bc03e13a2 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Wed, 16 Aug 2017 18:02:33 +0200
Subject: [PATCH] Adds acceptance fraction to saved data

---
 pyfstat/mcmc_based_searches.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py
index 78dbd60..649e1c8 100644
--- a/pyfstat/mcmc_based_searches.py
+++ b/pyfstat/mcmc_based_searches.py
@@ -352,9 +352,12 @@ class MCMCSearch(core.BaseSearchClass):
                                total=nburn+nprod):
                 pass
 
+        self.mean_acceptance_fraction = np.mean(
+            sampler.acceptance_fraction, axis=1)
         logging.info("Mean acceptance fraction: {}"
-                     .format(np.mean(sampler.acceptance_fraction, axis=1)))
+                     .format(self.mean_acceptance_fraction))
         if self.ntemps > 1:
+            self.tswap_acceptance_fraction = sampler.tswap_acceptance_fraction
             logging.info("Tswap acceptance fraction: {}"
                          .format(sampler.tswap_acceptance_fraction))
         try:
-- 
GitLab