From f26d2f7bc60b67bc1010d0d501262c0688a29f56 Mon Sep 17 00:00:00 2001 From: Mark McMahon Date: Thu, 6 Jun 2024 16:18:47 +0200 Subject: [PATCH] Remove neural --- .../data_science/model_comparison/nodes.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/src/mridle/pipelines/data_science/model_comparison/nodes.py b/src/mridle/pipelines/data_science/model_comparison/nodes.py index 11f11258..1aa987d7 100644 --- 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