uncomment renaming columns part

ivadomed · Jun 7, 2024 · 9c52132 · 9c52132
1 parent 14c97cb
commit 9c52132
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/compute_metrics_reloaded.py b/compute_metrics_reloaded.py
@@ -252,9 +252,9 @@ def main():
     df_mean = (df.drop(columns=['reference', 'prediction', 'EmptyRef', 'EmptyPred']).groupby('label').
                agg(['mean', 'std']).reset_index())
 
-    # # 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)
+    # 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)
 
     # format output up to 3 decimal places
     df = df.round(3)