예제 #1
0
 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)
예제 #2
0
 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)