Commit df586f0a authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Improvements to plot_walkers

- Remove arbitrary dpi choices
- Set the xlimits to remove white space before and after
parent 13a171c2
...@@ -449,7 +449,7 @@ class MCMCSearch(core.BaseSearchClass): ...@@ -449,7 +449,7 @@ class MCMCSearch(core.BaseSearchClass):
**kwargs) **kwargs)
fig.tight_layout() fig.tight_layout()
fig.savefig('{}/{}_init_{}_walkers.png'.format( fig.savefig('{}/{}_init_{}_walkers.png'.format(
self.outdir, self.label, j), dpi=400) self.outdir, self.label, j))
p0 = self._get_new_p0(sampler) p0 = self._get_new_p0(sampler)
p0 = self._apply_corrections_to_p0(p0) p0 = self._apply_corrections_to_p0(p0)
...@@ -469,7 +469,7 @@ class MCMCSearch(core.BaseSearchClass): ...@@ -469,7 +469,7 @@ class MCMCSearch(core.BaseSearchClass):
nprod=nprod, **kwargs) nprod=nprod, **kwargs)
fig.tight_layout() fig.tight_layout()
fig.savefig('{}/{}_walkers.png'.format(self.outdir, self.label), fig.savefig('{}/{}_walkers.png'.format(self.outdir, self.label),
dpi=200) )
samples = sampler.chain[0, :, nburn:, :].reshape((-1, self.ndim)) samples = sampler.chain[0, :, nburn:, :].reshape((-1, self.ndim))
lnprobs = sampler.lnprobability[0, :, nburn:].reshape((-1)) lnprobs = sampler.lnprobability[0, :, nburn:].reshape((-1))
...@@ -960,6 +960,8 @@ class MCMCSearch(core.BaseSearchClass): ...@@ -960,6 +960,8 @@ class MCMCSearch(core.BaseSearchClass):
axes[i].plot(xoffset+idxs[burnin_idx:], axes[i].plot(xoffset+idxs[burnin_idx:],
cs[burnin_idx:]-subtractions[i], cs[burnin_idx:]-subtractions[i],
color="k", alpha=alpha, lw=lw) color="k", alpha=alpha, lw=lw)
axes[i].set_xlim(0, xoffset+idxs[-1])
if symbols: if symbols:
if subtractions[i] == 0: if subtractions[i] == 0:
axes[i].set_ylabel(symbols[i], labelpad=labelpad) axes[i].set_ylabel(symbols[i], labelpad=labelpad)
...@@ -2105,7 +2107,7 @@ class MCMCFollowUpSearch(MCMCSemiCoherentSearch): ...@@ -2105,7 +2107,7 @@ class MCMCFollowUpSearch(MCMCSemiCoherentSearch):
return fig, axes return fig, axes
else: else:
fig.savefig('{}/{}_walkers.png'.format( fig.savefig('{}/{}_walkers.png'.format(
self.outdir, self.label), dpi=200) self.outdir, self.label))
class MCMCTransientSearch(MCMCSearch): class MCMCTransientSearch(MCMCSearch):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment