Python DecisionTree.getClassifier示例

示例#1

文件： DecisionTreeTest.py 项目： charanpald/sandbox

    def testMinSplit(self):
        decisionTree = DecisionTree()
        decisionTree.setMinSplit(20)
        decisionTree.learnModel(self.X, self.y)

        size = orngTree.countNodes(decisionTree.getClassifier())
        #orngTree.printTree(decisionTree.getClassifier())

        decisionTree.setMinSplit(0)
        decisionTree.learnModel(self.X, self.y)
        size2 = orngTree.countNodes(decisionTree.getClassifier())
        #orngTree.printTree(decisionTree.getClassifier())

        self.assertTrue(size < size2)

示例#2

文件： DecisionTreeTest.py 项目： charanpald/sandbox

    def testSetMaxDepth(self):
        maxDepth = 20
        decisionTree = DecisionTree()
        decisionTree.setMaxDepth(maxDepth)
        decisionTree.learnModel(self.X, self.y)

        self.assertTrue(DecisionTree.depth(decisionTree.getClassifier().tree) <= maxDepth+1)

        maxDepth = 5
        decisionTree = DecisionTree()
        decisionTree.setMaxDepth(maxDepth)
        decisionTree.learnModel(self.X, self.y)

        self.assertTrue(DecisionTree.depth(decisionTree.getClassifier().tree) <= maxDepth+1)

示例#3

文件： DecisionTreeTest.py 项目： rezaarmand/sandbox

    def testMinSplit(self):
        decisionTree = DecisionTree()
        decisionTree.setMinSplit(20)
        decisionTree.learnModel(self.X, self.y)

        size = orngTree.countNodes(decisionTree.getClassifier())
        #orngTree.printTree(decisionTree.getClassifier())

        decisionTree.setMinSplit(0)
        decisionTree.learnModel(self.X, self.y)
        size2 = orngTree.countNodes(decisionTree.getClassifier())
        #orngTree.printTree(decisionTree.getClassifier())

        self.assertTrue(size < size2)

示例#4

文件： DecisionTreeTest.py 项目： rezaarmand/sandbox

    def testSetMaxDepth(self):
        maxDepth = 20
        decisionTree = DecisionTree()
        decisionTree.setMaxDepth(maxDepth)
        decisionTree.learnModel(self.X, self.y)

        self.assertTrue(
            DecisionTree.depth(decisionTree.getClassifier().tree) <= maxDepth +
            1)

        maxDepth = 5
        decisionTree = DecisionTree()
        decisionTree.setMaxDepth(maxDepth)
        decisionTree.learnModel(self.X, self.y)

        self.assertTrue(
            DecisionTree.depth(decisionTree.getClassifier().tree) <= maxDepth +
            1)

示例#5

文件： DecisionTreeTest.py 项目： charanpald/sandbox

    def testLearnModel(self):
        decisionTree = DecisionTree()
        decisionTree.learnModel(self.X, self.y)

        tree = decisionTree.getClassifier()

示例#6

文件： DecisionTreeTest.py 项目： rezaarmand/sandbox

    def testLearnModel(self):
        decisionTree = DecisionTree()
        decisionTree.learnModel(self.X, self.y)

        tree = decisionTree.getClassifier()