Remove neural

uzh-dqbm-cmi · Jun 6, 2024 · f26d2f7 · f26d2f7
1 parent eaf452c
commit f26d2f7
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/src/mridle/pipelines/data_science/model_comparison/nodes.py b/src/mridle/pipelines/data_science/model_comparison/nodes.py
@@ -116,13 +116,6 @@ def create_model_precision_comparison_plot(evaluation_table_df: pd.DataFrame) ->
         altair scatter plot with number of positive predictions on the X-axis, and the precision on the Y-axis
 
     """
-    chart_config = alt.configure_axis(
-        labelFontSize=12,  # Axis label font size
-        titleFontSize=14  # Axis title font size
-    ).configure_legend(
-        labelFontSize=12,  # Legend label font size
-        titleFontSize=14  # Legend title font size
-    )
     model_precision_comparison_plot = alt.Chart(evaluation_table_df[['# No-show predictions per week',
                                                                      'PPV / Precision',
                                                                      'Model'
@@ -147,7 +140,13 @@ def create_model_precision_comparison_plot(evaluation_table_df: pd.DataFrame) ->
         y='PPV / Precision',
         color='Model'
     )
-    final_plot = (model_precision_comparison_plot + line_plot).apply(chart_config)
+    final_plot = (model_precision_comparison_plot + line_plot).configure_axis(
+        labelFontSize=12,  # Axis label font size
+        titleFontSize=14   # Axis title font size
+    ).configure_legend(
+        labelFontSize=12,  # Legend label font size
+        titleFontSize=14   # Legend title font size
+    )
 
     return final_plot