mirror of
https://github.com/Doctorado-ML/Odte.git
synced 2025-07-11 00:02:30 +00:00
Change default n_estimators to 10 instead of 100
This commit is contained in:
parent
382a420791
commit
f9b83adfee
@ -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__(
|
||||
|
@ -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(
|
||||
|
Loading…
x
Reference in New Issue
Block a user