Update src/vis_utils.py

src/vis_utils.py

@@ -179,31 +179,37 @@ def plot_family_results(method_names, dataset, metric, family_path="/tmp/family_
 
     # Filter by method names and selected dataset columns
     df = df[df['Method'].isin(method_names)]
-    metric_columns = [col for col in df.columns if col.startswith(f"{dataset}_{metric}_")]
-    
-    # Reshape data for plotting
-    df_long = pd.melt(df[['Method'] + metric_columns], id_vars=['Method'], var_name='Fold', value_name='Value')
-    df_long['Fold'] = df_long['Fold'].apply(lambda x: int(x.split('_')[-1]))  # Extract fold index
 
-    df = df.fillna(0)
+    # Filter columns based on the aspect and metrics
+    value_vars = [col for col in df.columns if col.startswith(f"{dataset}_") and "_" in col]
+
+    # Reshape the DataFrame to long format
+    df_long = pd.melt(df, id_vars=["Method"], value_vars=value_vars, var_name="Aspect_Metric_Fold", value_name="Value")
+
+    # Split the "Aspect_Metric_Fold" column into "Metric" and "Fold"
+    df_long[["Metric", "Fold"]] = df_long["Aspect_Metric_Fold"].str[len(aspect) + 1:].str.split("_", expand=True)
+    df_long["Fold"] = df_long["Fold"].astype(int)
 
     # Set up the plot
-    sns.set(rc={'figure.figsize': (13.7, 18.27)})
+    sns.set(rc={"figure.figsize": (13.7, 18.27)})
     sns.set_theme(style="whitegrid", color_codes=True)
-    ax = sns.boxplot(data=df_long, x='Value', y='Method', hue='Fold', whis=np.inf, orient="h")
-    
-    # Customize x-axis and y-axis tickers and grid
+
+    # Create boxplot
+    ax = sns.boxplot(data=df_long, x="Value", y="Method", hue="Metric", whis=np.inf, orient="h")
+
+    # Customize grid and ticks
     ax.xaxis.set_major_locator(ticker.MultipleLocator(0.2))
-    ax.get_xaxis().set_minor_locator(ticker.AutoMinorLocator())
-    ax.get_yaxis().set_minor_locator(ticker.AutoMinorLocator())
-    ax.grid(visible=True, which='major', color='gainsboro', linewidth=1.0)
-    ax.grid(visible=True, which='minor', color='whitesmoke', linewidth=0.5)
+    ax.xaxis.set_minor_locator(ticker.AutoMinorLocator())
+    ax.yaxis.set_minor_locator(ticker.AutoMinorLocator())
+    ax.grid(visible=True, which="major", color="gainsboro", linewidth=1.0)
+    ax.grid(visible=True, which="minor", color="whitesmoke", linewidth=0.5)
     ax.set_xlim(0, 1)
 
-    # Draw dashed lines between different representations on y-axis
+    # Add dashed lines between methods
     yticks = ax.get_yticks()
     for ytick in yticks:
-        ax.hlines(ytick + 0.5, -0.1, 1, linestyles='dashed')
+        ax.hlines(ytick + 0.5, -0.1, 1, linestyles="dashed", color="gray")
+
 
     # Apply color settings to y-axis labels
     for label in ax.get_yticklabels():