Update src/vis_utils.py

src/vis_utils.py

@@ -78,10 +78,10 @@ def plot_similarity_results(methods_selected, x_metric, y_metric, similarity_pat
     similarity_df = pd.read_csv(similarity_path)
     
     # Filter the dataframe based on selected methods
-    filtered_df = similarity_df[similarity_df['method_name'].isin(methods_selected)]
+    filtered_df = similarity_df[similarity_df['Method'].isin(methods_selected)]
 
     # Add a new column to the dataframe for the color
-    filtered_df['color'] = filtered_df['method_name'].apply(get_method_color)
+    filtered_df['color'] = filtered_df['Method'].apply(get_method_color)
 
     title, x_label, y_label = generate_labels_and_title(x_metric, y_metric)
 
@@ -97,7 +97,7 @@ def plot_similarity_results(methods_selected, x_metric, y_metric, similarity_pat
                    mapping=p9.aes(x=x_metric,  # Use the selected x_metric
                                   y=y_metric,  # Use the selected y_metric
                                   color='color',  # Use the dynamically generated color
-                                  label='method_name'))  # Label each point by the method name
+                                  label='Method'))  # Label each point by the method name
          + p9.geom_point(size=3)  # Add points with no jitter, set point size
          + p9.geom_text(nudge_y=0.02, size=8)  # Add method names as labels, nudge slightly above the points
          + p9.labs(title=title, x=x_label, y=y_label)  # Dynamic labels for X and Y axes