Merge pull request #46 from AllenNeuralDynamics/han_add_logistic_regr…

…ession_plot

feat: add logistic regression fits

code/Home.py

@@ -207,7 +207,7 @@ def draw_session_plots(df_to_draw_session):
 
     # Setting up layout for each session
     layout_definition = [[1],   # columns in the first row
-                         [1.5, 1],  # columns in the second row
+                         [1, 1],  # columns in the second row
                          [1, 1],
                          ]  
 
@@ -464,10 +464,15 @@ def init():
                                 root='foraging_auto_training/')
     )
 
+    logistic_regression_models = ['Su2022', 'Bari2019', 'Hattori2019', 'Miller2021']
+
 
     st.session_state.draw_type_mapper_session_level = {'1. Choice history': ('choice_history',   # prefix
                                                             (0, 0),     # location (row_idx, column_idx)
                                                             dict()),
+                                                       **{f'{n + 2}. Logistic regression ({model})': (f'logistic_regression_{model}',   # prefix
+                                                            (1 + int(n/2), n%2),     # location (row_idx, column_idx)
+                                                            dict()) for n, model in enumerate(logistic_regression_models)},
 
                                         # '1. Choice history': ('fitted_choice',   # prefix
                                         #                     (0, 0),     # location (row_idx, column_idx)