diff --git a/peyote/sampler.py b/peyote/sampler.py index 92e8b33c0f3c3bffc164113168cc406d1c6873c4..92d67028ca298310431465709499a6606c67321f 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])