Ejemplo n.º 1
0
    def testMinSplit(self):
        decisionTree = DecisionTree()
        decisionTree.setMinSplit(20)
        decisionTree.learnModel(self.X, self.y)

        size = decisionTree.getTree().node_count
        #orngTree.printTree(decisionTree.getClassifier())

        decisionTree.setMinSplit(1)
        decisionTree.learnModel(self.X, self.y)
        size2 = decisionTree.getTree().node_count
        #orngTree.printTree(decisionTree.getClassifier())

        self.assertTrue(size < size2)