swarm_boxplot(ax_10, model_exp1, 'q', 'Parameter estimate', 1)
ax_10.set_title('Reward bias')
# sigma_H
plt.subplot(224)
ax_11 = plt.gca()
swarm_boxplot(ax_11, model_exp1, 'sigma_H', 'Parameter estimate', 1)
ax_11.set_title('Catch trial')
# -------------------------------------
# 4. Add subplot labels and save figure
# -------------------------------------
# Adjust space and axes
plt.subplots_adjust(left=None,
bottom=None,
right=None,
top=None,
wspace=0.5,
hspace=0.6)
sns.despine()
texts = ['a', 'b', 'c', 'd']
label_subplots(f, texts, x_offset=0.08, y_offset=0.0)
savename = "/" + home_dir + "/rasmus/Dropbox/Apps/Overleaf/al_manuscript/al_figures/al_SM_figure_2.pdf"
plt.savefig(savename, transparent=True, dpi=400)
# Show figure
plt.show()
param_recov['sigma_H'].values,
'.',
color='k')
plt.title('Catch-trial')
plt.xlim(-1, 33)
plt.ylim(-1, 33)
plt.xlabel('True parameter')
plt.ylabel('Recovered parameter')
# -------------------------------------
# 4. Add subplot labels and save figure
# -------------------------------------
# Adjust space and axes
plt.subplots_adjust(left=None,
bottom=None,
right=None,
top=None,
wspace=0.5,
hspace=0.5)
sns.despine()
texts = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i'] # label letters
label_subplots(f, texts, x_offset=0.07, y_offset=0.01)
savename = "/" + home_dir + "/rasmus/Dropbox/Apps/Overleaf/al_manuscript/al_figures/al_SM_figure_8.pdf"
plt.savefig(savename)
# Show figure
plt.show()
