wspace=0.3,
hspace=0.3,
left=0.1,
right=0.9,
bottom=0.06,
top=0.94)
base_a = util.panel_label(fig, fine_gc[0, 0], "a")
base_b = util.panel_label(fig, fine_gc[0, 2], "b")
base_c = util.panel_label(fig, gs[1, 0], "c")
# ---------------- readout panel ----
# panel a
ax_b = fig.add_subplot(fine_gc[0, 2])
gain_mappable = readouts.gain_map(ax_b, readout_data)
util.labels(ax_b, "Layer 4 Feature Map", "")
util.colorbar(
fig,
ax_b,
gain_mappable,
label=pkws.labels.effective_gain,
shrink=0,
)
# panel b
ax_c = fig.add_subplot(gs[1, 0])
readouts.reconstructed_bhv(ax_c, readout_data, regs)
util.labels(ax_c, None, pkws.labels.bhv_performance)
# save
plt.savefig(params.output, transparent=True)
color=pkws.pal_bhv[1],
zorder=3,
**pkws.bhv_mean)
ax.set_xlim(-0.5, 3.5)
# ax.axhline(0.5, lw = 1, color = '.7', zorder = -1, ls = '--')
ax.set_xticks([0, 1, 2])
ax.set_ylim(pkws.bhv_yrng)
if yticks is None:
ax.set_yticks(ax.get_yticks())
else:
ax.set_yticks(yticks)
# panel b
ax_c = fig.add_subplot(gs[0, 0])
reconstructed_bhv(ax_c, readout_data, regs, pkws=pkws, yticks=pkws.bhv_yrng)
ax_c.set_xticks([0, 1, 2])
sns.despine(ax=ax_c, trim=True, offset=5)
ax_c.set_xticklabels([
pkws.labels.reconst_models[0], pkws.labels.reconst_models[2],
pkws.labels.reconst_models[3]
],
rotation=45,
ha='center')
ax_c.set_xlim(-0.5, 2.5)
util.labels(ax_c, None, "Performance [AUC (d')]", pkws=pkws)
# save
plt.savefig(params.output, transparent=True)
# make gridspec
gs = gridspec.GridSpec(
nrows = 1, ncols = 1, figure = fig,)
ax_a = fig.add_subplot(gs[0, 0])
diagrams.rf_ellipses(
ax_a, lp_pre_ells, lp_att_ells[3],
grads_dist, grads_focl, (0,4,0),
params.rf_diagram_units, loc = (56, 56),
color_dist = '#000000', color_focl = '#d55c00', pkws = pkws)
ax_a.set_ylim([150, 0])
util.axis_expand(ax_a, L = 0.0, B = 0.05, R = 0.05, T = 0.1)
util.labels(ax_a,
"Image space",
"Image space", pkws = pkws)
# save
plt.savefig(params.output, transparent = True)
pal=pkws.pal_b,
xlim=pkws.lineplot_xlim,
ylim=params.shift_lim,
xticks=[0, 150],
pkws=pkws)
for ax in ax_c:
ax.set_yticks([ax.get_yticks()[0], ax.get_yticks()[-1]])
ax.set_yticklabels(["", ""])
ax.tick_params(length=pkws.rc['ytick.major.size'] / 2)
ax_c[-1].set_xticklabels(["", ""])
# c: main panel
ax_c = fig.add_subplot(gs_c[:, 0])
lineplots.lineplot(lineplots.rf_file_iterator('shift',
lp_dists,
lp_att_ells, (0, 4, 0),
comp_ells=lp_comp_ells),
ax_c,
line_span=pkws.lineplot_span,
rad=30,
pal=pkws.pal_b,
xlim=pkws.lineplot_xlim,
ylim=params.shift_lim,
pkws=pkws)
ax_c.set_yticks([params.shift_lim[0], 0, params.shift_lim[1]])
ax_c.set_xticks([0, 150])
sns.despine(ax=ax_c, trim=True, offset=5)
util.labels(ax_c, pkws.labels.unit_distance, pkws.labels.rf_shift, pkws=pkws)
# save
plt.savefig(params.output, transparent=True)
figure=fig,
)
ax_e = fig.add_subplot(gs[0, 0])
lineplots.lineplot(lineplots.gain_file_iterator(lp_dists,
sgain_focl, (0, 4, 0),
gain_comp=sgain_comp),
ax_e,
line_span=pkws.lineplot_span,
rad=30,
pal=pkws.pal_b,
xlim=(0, 180),
ylim=params.gain_lim,
pkws=pkws)
util.labels(ax_e,
pkws.labels.unit_distance,
pkws.labels.effective_gain,
pkws=pkws)
util.legend(fig,
ax_e,
pkws.labels.beta,
pkws.pal_b,
inset=pkws.legend_inset,
pkws=pkws)
ax_e.set_yticks([1, params.gain_lim[1]])
ax_e.set_xticks([0, 150])
sns.despine(ax=ax_e, trim=True, offset=5)
util.axis_expand(ax_e, L=-0.07, B=-0.1, R=0.1, T=0.05)
# save
plt.savefig(params.output, transparent=True)
nrows=1,
ncols=1,
figure=fig,
)
ax_d = fig.add_subplot(gs[0, 0])
size_map = quivers.quiverplot(qv_dist_ell,
qv_focl_ell,
qv_smooth_samp,
ax_d,
cmap='coolwarm',
vrng=params.size_lim,
pkws=pkws)
util.axis_expand(ax_d, L=0.15, B=0.1, R=-0.11, T=0.15)
util.labels(ax_d,
pkws.labels.image_position.format('Horizontal'),
pkws.labels.image_position.format('Vertical'),
pkws=pkws)
util.colorbar(fig,
ax_d,
size_map,
ticks=params.size_lim + (1, ),
label=pkws.labels.rf_size,
label_vofs=-0.12,
label_margin=0.04,
margin=0.03,
width=0.03,
pkws=pkws)
# save
plt.savefig(params.output, transparent=True)
# pkws.labels.image_position.format('Vertical'))
# util.colorbar(
# fig, ax_d, size_map, ticks = params.size_lim + (1,),
# label = pkws.labels.rf_size, label_vofs = -0.03)
# panel c : single axis
ax_c = fig.add_subplot(gs[1,1])
lineplots.lineplot(
lineplots.rf_file_iterator(
'shift', lp_dists, lp_att_ells, (0,4,0),
comp_ells = lp_comp_ells),
ax_c,
line_span = 30, rad = 30, pal = pkws.pal_b,
xlim = (0, 180), ylim = params.shift_lim)
util.labels(ax_c, pkws.labels.unit_distance, pkws.labels.rf_shift)
util.legend(
fig, ax_c, pkws.labels.beta, pkws.pal_b,
inset = pkws.legend_inset)
# panel d : single axis
ax_d = fig.add_subplot(gs[1,2])
lineplots.lineplot(
lineplots.rf_file_iterator(
'size', lp_dists, lp_att_ells, (0,4,0),
comp_ells = lp_comp_ells),
ax_d,
line_span = 30, rad = 30, pal = pkws.pal_b,
xlim = (0, 180), ylim = params.size_lim)
util.labels(ax_d, pkws.labels.unit_distance, pkws.labels.rf_size)
base_b = util.panel_label(fig, gs[1, 0], "b")
base_c = util.panel_label(fig, gs[1, 1], "c")
base_d = util.panel_label(fig, gs[1, 2], "d")
base_e = util.panel_label(fig, gs[2, 0], "e")
base_f = util.panel_label(fig, gs[2, 1], "f")
# panel b
ax_d = fig.add_subplot(gs[1, 0])
size_map = quivers.quiverplot(qv_dist_ell,
qv_focl_ell,
qv_smooth_samp,
ax_d,
cmap='coolwarm',
vrng=params.size_lim)
util.axis_expand(ax_d, L=0.2, B=0.2, R=-0.1, T=0.05)
util.labels(ax_d, pkws.labels.image_position.format('Horizontal'),
pkws.labels.image_position.format('Vertical'))
util.colorbar(fig,
ax_d,
size_map,
ticks=params.size_lim + (1, ),
label=pkws.labels.rf_size,
label_vofs=-0.04)
# panel c : single axis
ax_c = fig.add_subplot(gs[1, 1])
lineplots.lineplot(lineplots.rf_file_iterator('shift',
lp_dists,
lp_att_ells, (0, 4, 0),
comp_ells=lp_comp_ells),
ax_c,
line_span=30,
