end Decision tree

pages/application/DecisionTree/utils/dtviz.py

@@ -23,6 +23,8 @@ def create_legend(g):
     legend.add_node("b", style = "invis")
     legend.add_node("c", style = "invis")
     legend.add_node("d", style = "invis")
+    legend.add_node("e", style = "invis")
+    legend.add_node("f", style = "invis")
 
     legend.add_edge("a","b")
     edge = legend.get_edge("a","b")
@@ -35,7 +37,10 @@ def create_legend(g):
     edge.attr["color"] = "blue"
     edge.attr["style"] = "dashed"
 
-
+    legend.add_edge("e","f")  
+    edge = legend.get_edge("e","f")
+    edge.attr["label"] = "contrastive explanation"
+    edge.attr["color"] = "red"
 #
 #==============================================================================
 def visualize(dt):