Change n_estimators to sensible default value

chkoar · Jan 3, 2019 · f41787d · f41787d
1 parent d9f7260
commit f41787d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/dagging/_dagging.py b/dagging/_dagging.py
@@ -13,7 +13,7 @@
 class BaseDagging(BaseEnsemble):
     def __init__(self,
                  base_estimator=None,
-                 n_estimators=3,
+                 n_estimators=10,
                  voting='soft',
                  random_state=None):
         super(BaseDagging, self).__init__(
@@ -114,7 +114,7 @@ class DaggingClassifier(BaseDagging, ClassifierMixin):
 
     def __init__(self,
                  base_estimator=None,
-                 n_estimators=3,
+                 n_estimators=10,
                  voting='soft',
                  random_state=None):
         super(DaggingClassifier, self).__init__(
@@ -225,7 +225,7 @@ class DaggingRegressor(BaseDagging, RegressorMixin):
 
     def __init__(self,
                  base_estimator=None,
-                 n_estimators=3,
+                 n_estimators=10,
                  random_state=None):
         super(DaggingRegressor, self).__init__(
             base_estimator=base_estimator,

diff --git a/dagging/tests/test_dagging.py b/dagging/tests/test_dagging.py
@@ -9,5 +9,5 @@ def test_check_dagging_classifier():
 
 
 def test_check_dagging_regressor():
-    model = DaggingRegressor(random_state=0)
+    model = DaggingRegressor(random_state=0, n_estimators=2)
     check_estimator(model)