From 02257a5901ce09e4f58d68ab68ef143af43b8f5c Mon Sep 17 00:00:00 2001
From: bbauvin <baptiste.bauvin@centrale-marseille.fr>
Date: Mon, 2 Oct 2017 17:31:39 -0400
Subject: [PATCH] Modified error in name generation for final graph

---
 Code/MonoMutliViewClassifiers/ResultAnalysis.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/Code/MonoMutliViewClassifiers/ResultAnalysis.py b/Code/MonoMutliViewClassifiers/ResultAnalysis.py
index 3f9f9028..e092fdd8 100644
--- a/Code/MonoMutliViewClassifiers/ResultAnalysis.py
+++ b/Code/MonoMutliViewClassifiers/ResultAnalysis.py
@@ -28,11 +28,16 @@ def autolabel(rects, ax):
                 ha='center', va='bottom')
 
 
+def genFusionName(type_, a, b, c):
+    if type_ == "Fusion" and a["fusionType"] != "EarlyFusion":
+        return "Late-"+str(a["fusionMethod"])
+    elif type_ == "Fusion" and a["fusionType"] != "LateFusion":
+        return "Early-"+a["fusionMethod"]+"-"+a["classifiersNames"][0]
+
+
 def genNamesFromRes(mono, multi):
     names = [res[1][0]+"-"+res[1][1][-1] for res in mono]
-    names+=[type_ for type_, a, b, c in multi if type_ != "Fusion"]
-    names+=[ "Late-"+str(a["fusionMethod"]) for type_, a, b, c in multi if type_ == "Fusion" and a["fusionType"] != "EarlyFusion"]
-    names+=[ "Early-"+a["fusionMethod"]+"-"+a["classifiersNames"][0]  for type_, a, b, c in multi if type_ == "Fusion" and a["fusionType"] != "LateFusion"]
+    names += [type_ if type_ != "Fusion" else genFusionName(type_, a, b, c) for type_, a, b, c in multi]
     return names
 
 
-- 
GitLab