Figure S6: Neural representations with biological plasticity rules

from EIANN.generate_figures import *
import EIANN as eiann
eiann.update_plot_defaults()
import nbimporter
import Figure_2 as fig2

figure_name = "FigS6_biological_representations"
model_list = ["BTSP_WT_hebbdend", "Supervised_HebbWN_WT_hebbdend",
            "Supervised_BCM_WT_hebbdend", "SupHebbTempCont_WT_hebbdend"]

model_dict_all = load_model_dict()
generate_hdf5_all_seeds(model_list, model_dict_all, variables_to_save=["average_pop_activity_dict", "metrics_dict"], recompute=None)

model_dict_all["SupHebbTempCont_WT_hebbdend"]["display_name"] = "Temp. Contr. Hebb"
model_dict_all["Supervised_HebbWN_WT_hebbdend"]["display_name"] = "Hebb + W Norm."
model_dict_all["Supervised_HebbWN_WT_hebbdend"]["label"] = "Hebb + W Norm."

fig = fig2.plot_E_pop_representations(model_list, model_dict_all, figure_name, plot_accuracy=False, plot_confusion=False)
fig.suptitle("Biological plasticity rules under Dend. Target Prop.", fontsize=9, y=1.)
root_dir = ut.get_project_root()
fig.savefig(f"{root_dir}/EIANN/figures/{figure_name}.svg", dpi=300)
fig.savefig(f"{root_dir}/EIANN/figures/{figure_name}.png", dpi=300)