Figure S7: Receptive fields with biological plasticity

from EIANN.generate_figures import *
import EIANN as eiann
eiann.update_plot_defaults()
import nbimporter
import Figure_5 as fig5

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

model_dict_all = load_model_dict()
model_dict_all["Supervised_HebbWN_WT_hebbdend"]["display_name"] = "Hebb + W Norm."
model_dict_all["SupHebbTempCont_WT_hebbdend"]["display_name"] = "Temp Contrastive Hebb"
generate_hdf5_all_seeds(model_list, model_dict_all, variables_to_save=["metrics_dict", "final_receptive_fields", "average_pop_activity_dict"], recompute=None)

population = 'H1E'
fig5.generate_receptive_field_figure(population, model_list, model_dict_all, figure_name)

population = 'H2E'
fig5.generate_receptive_field_figure(population, model_list, model_dict_all, figure_name)