Spaces:
Running
Running
Update src/vis_utils.py
Browse files- 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 |
|