Commit 5ccd2e3e authored by Gregory Ashton's avatar Gregory Ashton

Simplify projection matrix plots

parent 9c2a9c57
Pipeline #3855 passed with stage
in 10 seconds
......@@ -755,7 +755,8 @@ class EarthTest(GridSearch):
F_at_zero, marginalised_F, max_F, max_F_params))
return F_at_zero - marginalised_F, (F_at_zero - max_F) / F_at_zero
def plot_corner(self, prior_widths=None, fig=None, axes=None):
def plot_corner(self, prior_widths=None, fig=None, axes=None,
projection='log_mean'):
Bsa, FmaxMismatch = self.marginalised_bayes_factor(prior_widths)
data = self.data[:, -1].reshape(
......@@ -770,7 +771,7 @@ class EarthTest(GridSearch):
from projection_matrix import projection_matrix
fig, axes = projection_matrix(data, xyz, projection='log_mean',
fig, axes = projection_matrix(data, xyz, projection=projection,
factor=1.6, labels=labels)
axes[-1][-1].axvline((lal.DAYJUL_SI - lal.DAYSID_SI)/60.0, color='C3')
plt.suptitle(
......@@ -782,13 +783,6 @@ class EarthTest(GridSearch):
fig.savefig('{}/{}_projection_matrix.png'.format(
self.outdir, self.label))
fig, axes = projection_matrix(data, xyz, projection='max_slice',
factor=1.6, labels=labels)
axes[-1][-1].axvline((lal.DAYJUL_SI - lal.DAYSID_SI)/60.0, color='C3')
plt.suptitle('T={:.1f} days, $f$={:.2f} Hz' .format(
self.duration/86400, self.F0), y=0.99, size=14)
fig.savefig('{}/{}_max_slice.png'.format(self.outdir, self.label))
def plot(self, key, prior_widths=None):
Bsa, FmaxMismatch = self.marginalised_bayes_factor(prior_widths)
......
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