From 227fc983c53db406499cf82ddd3d49b4ac668322 Mon Sep 17 00:00:00 2001 From: xuyuon <116078673+xuyuon@users.noreply.github.com> Date: Thu, 25 Jul 2024 08:59:32 -0400 Subject: [PATCH] Updated runManager.py --- src/jimgw/single_event/runManager.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/jimgw/single_event/runManager.py b/src/jimgw/single_event/runManager.py index db90dd77..124e9de5 100644 --- a/src/jimgw/single_event/runManager.py +++ b/src/jimgw/single_event/runManager.py @@ -366,7 +366,7 @@ def sample(self): def get_samples(self): return self.jim.get_samples() - def plot_corner(self, path: str="corner.png", **kwargs): + def plot_corner(self, path: str="corner.jpeg", **kwargs): """ plot corner plot of the samples. """ @@ -383,7 +383,7 @@ def plot_corner(self, path: str="corner.png", **kwargs): plt.savefig(path) plt.close() - def plot_diagnostic(self, path: str="diagnostic.png", **kwargs): + def plot_diagnostic(self, path: str="diagnostic.jpeg", **kwargs): """ plot diagnostic plot of the samples. """ @@ -395,8 +395,7 @@ def plot_diagnostic(self, path: str="diagnostic.png", **kwargs): axs = [plt.subplot(2, 2, i + 1) for i in range(4)] plt.sca(axs[0]) plt.title("log probability") - for i in range(log_prob.shape[0]): - axs[0].plot(log_prob[i], linewidth=0.05) + plt.plot(log_prob.mean(0)) plt.xlabel("iteration") plt.sca(axs[1])