From e34da01852015b456842c74a8c2de67756c2d5ee Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Fri, 1 May 2020 08:23:22 -0400
Subject: [PATCH] Metric files renaming

---
 .../result_analysis/metric_analysis.py               | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/summit/multiview_platform/result_analysis/metric_analysis.py b/summit/multiview_platform/result_analysis/metric_analysis.py
index 3b9e4b08..c2db7b26 100644
--- a/summit/multiview_platform/result_analysis/metric_analysis.py
+++ b/summit/multiview_platform/result_analysis/metric_analysis.py
@@ -109,8 +109,7 @@ def publish_metrics_graphs(metrics_scores, directory, database_name,
                            nb_results, metric_name, file_name,
                            tag=" " + " vs ".join(labels_names))
 
-        class_file_name = os.path.join(directory, database_name + "-"
-                                       + metric_name + "-class")
+        class_file_name = file_name+"-class"
         plot_class_metric_scores(class_test_scores, class_file_name,
                                  labels_names, classifier_names, metric_name)
         logging.debug(
@@ -125,6 +124,8 @@ def publish_all_metrics_scores(iter_results, class_iter_results, directory,
     secure_file_path(os.path.join(directory, "a"))
 
     for metric_name, scores in iter_results.items():
+        if metric_name.endswith("*"):
+            metric_name = metric_name[:-1]+"_p"
         train = np.array(scores["mean"].loc["train"])
         test = np.array(scores["mean"].loc["test"])
         classifier_names = np.array(scores["mean"].columns)
@@ -168,7 +169,12 @@ def init_plot(results, metric_name, metric_dataframe,
 
     nb_results = metric_dataframe.shape[1]
 
-    file_name = os.path.join(directory, database_name + "-" + metric_name)
+    if metric_name.endswith("*"):
+        formatted_metric_name = metric_name[:-1]+"_p"
+    else:
+        formatted_metric_name = metric_name
+
+    file_name = os.path.join(directory, database_name + "-" + formatted_metric_name)
 
     results += [[classifiers_name, metric_name, test_mean, test_std, class_mean]
                 for classifiers_name, test_mean, class_mean, test_std in
-- 
GitLab