classDecsionNode.py

class DecisionNode(Node):
    def __init__(self,label,distr,threshold ,left = None,right = None):
        Node.__init__(self,label,left,right)
        self.threshold = threshold
        self.distr = distr
    def __str__(self):
        L = self.linearize()
        s = ""
        for e in L:
            s += "IF "
            for i in range(1,len(e)):
                if e[i] == e[i-1].left:
                    s += "{} >= {} AND ".format(e[i-1].label, str(e[i-1].threshold))
                else:
                    s += "{} < {} AND ".format(e[i-1].label,str(e[i-1].threshold))
            s = "{} THEN {} = {} \n".format(s[:len(s) -4],e[len(e)-1].label,str(e[len(e)-1].distr))
        return s