diff --git a/odte/Odte.py b/odte/Odte.py index c94afbb..9b9f26c 100644 --- a/odte/Odte.py +++ b/odte/Odte.py @@ -35,7 +35,7 @@ class Odte(BaseEnsemble, ClassifierMixin): random_state: int = 0, max_features: Optional[Union[str, int, float]] = None, max_samples: Optional[Union[int, float]] = None, - n_estimators: int = 100, + n_estimators: int = 10, be_hyperparams: str = "{}", ): super().__init__( diff --git a/odte/tests/Odte_tests.py b/odte/tests/Odte_tests.py index 498e9f2..baa572e 100644 --- a/odte/tests/Odte_tests.py +++ b/odte/tests/Odte_tests.py @@ -48,6 +48,7 @@ class Odte_test(unittest.TestCase): random_state=self._random_state, max_features=max_features, n_jobs=1, + n_estimators=100, ) tclf.fit(X, y) computed = tclf._get_random_subspace(X, y, tclf.max_features_) @@ -100,6 +101,7 @@ class Odte_test(unittest.TestCase): random_state=self._random_state, max_features=1.0, max_samples=0.1, + n_estimators=100, ) tclf.set_params( **dict(