mgyigit commited on
Commit
e281896
·
verified ·
1 Parent(s): 39bb5eb

Update src/vis_utils.py

Browse files
Files changed (1) hide show
  1. src/vis_utils.py +4 -1
src/vis_utils.py CHANGED
@@ -79,7 +79,10 @@ def plot_similarity_results(methods_selected, x_metric, y_metric, similarity_pat
79
 
80
  # Filter the dataframe based on selected methods
81
  filtered_df = similarity_df[similarity_df['Method'].isin(methods_selected)]
82
-
 
 
 
83
  # Add a new column to the dataframe for the color
84
  filtered_df['color'] = filtered_df['Method'].apply(get_method_color)
85
 
 
79
 
80
  # Filter the dataframe based on selected methods
81
  filtered_df = similarity_df[similarity_df['Method'].isin(methods_selected)]
82
+
83
+ # Replace None or NaN values with 0 in relevant columns
84
+ filtered_df = filtered_df.fillna(0)
85
+
86
  # Add a new column to the dataframe for the color
87
  filtered_df['color'] = filtered_df['Method'].apply(get_method_color)
88