convert multi-index cols to flat index

ivadomed · Jun 11, 2024 · 06f66b0 · 06f66b0
1 parent 0dd51ec
commit 06f66b0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/compute_metrics_reloaded.py b/compute_metrics_reloaded.py
@@ -259,6 +259,11 @@ def main():
     df_mean = (df.drop(columns=['reference', 'prediction', 'EmptyRef', 'EmptyPred']).groupby('label').
                agg(['mean', 'std']).reset_index())
 
+    # Convert multi-index to flat index
+    df_mean.columns = ['_'.join(col).strip() for col in df_mean.columns.values]
+    # Rename column `label_` back to `label`
+    df_mean.rename(columns={'label_': 'label'}, inplace=True)
+
     # Rename columns
     df.rename(columns={metric: METRICS_TO_NAME[metric] for metric in METRICS_TO_NAME}, inplace=True)
     df_mean.rename(columns={metric: METRICS_TO_NAME[metric] for metric in METRICS_TO_NAME}, inplace=True)