Ejemplo n.º 1
0
    def testGenerate(self):
        generate = DecisionTree.generate(5, 50)

        learner = generate()
        learner.learnModel(self.X, self.y)

        self.assertEquals(learner.getMaxDepth(), 5)
        self.assertEquals(learner.getMinSplit(), 50)