Skip to content
Snippets Groups Projects
Commit 29bc9c7a authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Remove detStat plot by default

parent 52c34f6e
Branches
No related tags found
No related merge requests found
...@@ -735,7 +735,7 @@ class MCMCSearch(core.BaseSearchClass): ...@@ -735,7 +735,7 @@ class MCMCSearch(core.BaseSearchClass):
def plot_walkers(self, sampler, symbols=None, alpha=0.4, color="k", temp=0, def plot_walkers(self, sampler, symbols=None, alpha=0.4, color="k", temp=0,
lw=0.1, nprod=0, add_det_stat_burnin=False, lw=0.1, nprod=0, add_det_stat_burnin=False,
fig=None, axes=None, xoffset=0, plot_det_stat=True, fig=None, axes=None, xoffset=0, plot_det_stat=False,
context='classic', subtractions=None, labelpad=0.05): context='classic', subtractions=None, labelpad=0.05):
""" Plot all the chains from a sampler """ """ Plot all the chains from a sampler """
...@@ -761,12 +761,16 @@ class MCMCSearch(core.BaseSearchClass): ...@@ -761,12 +761,16 @@ class MCMCSearch(core.BaseSearchClass):
if len(subtractions) != self.ndim: if len(subtractions) != self.ndim:
raise ValueError('subtractions must be of length ndim') raise ValueError('subtractions must be of length ndim')
if plot_det_stat:
extra_subplots = 1
else:
extra_subplots = 0
with plt.style.context((context)): with plt.style.context((context)):
plt.rcParams['text.usetex'] = True plt.rcParams['text.usetex'] = True
if fig is None and axes is None: if fig is None and axes is None:
fig = plt.figure(figsize=(4, 3.0*ndim)) fig = plt.figure(figsize=(4, 3.0*ndim))
ax = fig.add_subplot(ndim+1, 1, 1) ax = fig.add_subplot(ndim+extra_subplots, 1, 1)
axes = [ax] + [fig.add_subplot(ndim+1, 1, i) axes = [ax] + [fig.add_subplot(ndim+extra_subplots, 1, i)
for i in range(2, ndim+1)] for i in range(2, ndim+1)]
idxs = np.arange(chain.shape[1]) idxs = np.arange(chain.shape[1])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment