def train_random_forest(self, train_sizes, data_sets):
     logging.info("Training random forest on {} datasets.".format(
         len(data_sets)))
     self.random_forest = MyRandomForest(data_sets, self.dataset_map)
     self.random_forest.train(train_sizes)
Esempio n. 2
0
 def train_random_forest(self, train_sizes, data_sets):
     self.random_forest = MyRandomForest(data_sets, self.dataset_map,
                                         "model/lr_all.pkl")
     self.random_forest.train(train_sizes)