Change default n_estimators to 10 instead of 100

This commit is contained in:
Ricardo Montañana Gómez 2023-10-10 09:34:43 +02:00
parent 382a420791
commit f9b83adfee
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 3 additions and 1 deletions

View File

@ -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__(

View File

@ -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(