Commit 11f162d0 authored by Gregory Ashton's avatar Gregory Ashton

Minor polishing of corner plots

parent 78ecb25b
import numpy as np
import matplotlib.pyplot as plt
import pyfstat
import gridcorner
from make_simulated_data import tstart, duration, tref, F0, F1, F2, Alpha, Delta, delta_F0, dtglitch, outdir
plt.style.use('paper')
plt.style.use('./paper.mplstyle')
label = 'semicoherent_glitch_robust_directed_MCMC_search_on_1_glitch'
......@@ -47,5 +48,7 @@ mcmc.transform_dictionary['F1'] = dict(
subtractor=F1, symbol='$\dot{f}-\dot{f}^\mathrm{s}$')
mcmc.run()
mcmc.plot_corner(label_offset=0.15, truths=[0, 0, delta_F0, 50])
fig_and_axes = gridcorner._get_fig_and_axes(4, 2, 0.05)
mcmc.plot_corner(label_offset=0.35, truths=[0, 0, delta_F0, 50],
fig_and_axes=fig_and_axes)
mcmc.print_summary()
......@@ -12,7 +12,7 @@ except ImportError:
label = 'semicoherent_glitch_robust_directed_grid_search_on_1_glitch'
plt.style.use('paper')
plt.style.use('./paper.mplstyle')
Nstar = 1000
F0_width = np.sqrt(Nstar)*np.sqrt(12)/(np.pi*duration)
......@@ -29,6 +29,8 @@ tglitchs = [tstart+0.1*duration, tstart+0.9*duration, 0.8*float(duration)/N]
delta_F0s = [0, max_delta_F0, max_delta_F0/N]
delta_F1s = [0]
print 'Prior widths=', F0_width, F1_width
search = pyfstat.GridGlitchSearch(
label, outdir, 'data/*1_glitch*sft', F0s=F0s, F1s=F1s, F2s=F2s,
Alphas=Alphas, Deltas=Deltas, tref=tref, minStartTime=tstart,
......@@ -48,7 +50,7 @@ xyz = [F0_vals, F1_vals, delta_F0s_vals, tglitch_vals_days]
labels = ['$f - f^\mathrm{s}$\n[Hz]', '$\dot{f} - \dot{f}^\mathrm{s}$\n[Hz/s]',
'$\delta f$\n[Hz]', '$t^g_0$\n[days]', '$\widehat{2\mathcal{F}}$']
fig, axes = gridcorner(
twoF, xyz, projection='log_mean', whspace=0.1, factor=1.2, labels=labels,
showDvals=False, lines=[0, 0, delta_F0, 50])
twoF, xyz, projection='log_mean', labels=labels,
showDvals=False, lines=[0, 0, delta_F0, 50], label_offset=0.35)
fig.savefig('{}/{}_projection_matrix.png'.format(outdir, label),
bbox_inches='tight')
......@@ -743,7 +743,7 @@ class MCMCSearch(core.BaseSearchClass):
max_n_ticks=4,
plot_contours=True,
plot_datapoints=True,
label_kwargs={'fontsize': 12},
#label_kwargs={'fontsize': 12},
data_kwargs={'alpha': 0.1,
'ms': 0.5},
range=_range,
......@@ -762,10 +762,10 @@ class MCMCSearch(core.BaseSearchClass):
ax.set_rasterization_zorder(-10)
for tick in ax.xaxis.get_major_ticks():
tick.label.set_fontsize(8)
#tick.label.set_fontsize(8)
tick.label.set_rotation('horizontal')
for tick in ax.yaxis.get_major_ticks():
tick.label.set_fontsize(8)
#tick.label.set_fontsize(8)
tick.label.set_rotation('vertical')
plt.tight_layout(h_pad=0.0, w_pad=0.0)
......
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