def test_extra_trees_classifier(self): model = ExtraTreesClassifier(n_estimators=3) dump_binary_classification(model, folder=self.folder) dump_multiple_classification(model, folder=self.folder)
def test_random_forest_classifier(self): model = RandomForestClassifier(n_estimators=3) dump_binary_classification(model, folder=self.folder, nrows=2) dump_binary_classification(model, folder=self.folder) dump_multiple_classification(model, folder=self.folder)