Skip to content
Snippets Groups Projects
Commit 02257a59 authored by bbauvin's avatar bbauvin
Browse files

Modified error in name generation for final graph

parent ba8a026a
No related branches found
No related tags found
No related merge requests found
...@@ -28,11 +28,16 @@ def autolabel(rects, ax): ...@@ -28,11 +28,16 @@ def autolabel(rects, ax):
ha='center', va='bottom') 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): def genNamesFromRes(mono, multi):
names = [res[1][0]+"-"+res[1][1][-1] for res in mono] 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 += [type_ if type_ != "Fusion" else genFusionName(type_, a, b, c) for type_, a, b, c in multi]
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"]
return names return names
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment