Esempio n. 1
0
 def testprune(self): 
     learner = DecisionTreeLearner(minSplit=5)
     learner.learnModel(self.X, self.y)
     
     unprunedTree = learner.getTree().copy()
     
     learner.cartPrune(self.X, self.y)
 
     self.assertTrue(learner.tree.isSubtree(unprunedTree))