From 964f8c23cfaceba7cf2a845a8249d93516c2b397 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Tue, 1 May 2018 20:43:15 +1000
Subject: [PATCH] Further minor improvements to ptemcee implementation
---
peyote/sampler.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/peyote/sampler.py b/peyote/sampler.py
index 92e8b33c0..92d67028c 100644
--- a/peyote/sampler.py
+++ b/peyote/sampler.py
@@ -332,14 +332,14 @@ class Ptemcee(Sampler):
self.result.logz = np.nan
self.result.logzerr = np.nan
self.plot_walkers()
- logging.info("Autocorr time = {}".format(sampler.get_autocorr_time()))
+ logging.info("Max autocorr time = {}".format(np.max(sampler.get_autocorr_time())))
logging.info("Tswap frac = {}".format(sampler.tswap_acceptance_fraction))
return self.result
def plot_walkers(self, save=True, **kwargs):
nwalkers, nsteps, ndim = self.result.walkers.shape
idxs = np.arange(nsteps)
- fig, axes = plt.subplots(nrows=ndim)
+ fig, axes = plt.subplots(nrows=ndim, figsize=(6, 3*self.ndim))
for i, ax in enumerate(axes):
ax.plot(idxs, self.result.walkers[:, :, i].T, lw=0.1, color='k')
ax.set_ylabel(self.result.labels[i])
--
GitLab