def __init__(self, max_depth, num_trees):
     RandomTree.__init__(self, max_depth)
     self.num_trees = num_trees
Ejemplo n.º 2
0
 def __init__(self, max_depth, num_trees):
     self.num_trees = num_trees
     RandomTree.__init__(self, max_depth=max_depth)
     DecisionTree.__init__(self,
                           max_depth=max_depth,
                           stump_class=RandomStumpInfoGain)