diff --git a/dagging/_dagging.py b/dagging/_dagging.py index c6be2a0..f964e0c 100644 --- 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 index 2b009fc..371f8d9 100644 --- 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)