mirror of
https://github.com/Doctorado-ML/Odte.git
synced 2025-07-11 16:22:00 +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,
|
random_state: int = 0,
|
||||||
max_features: Optional[Union[str, int, float]] = None,
|
max_features: Optional[Union[str, int, float]] = None,
|
||||||
max_samples: Optional[Union[int, float]] = None,
|
max_samples: Optional[Union[int, float]] = None,
|
||||||
n_estimators: int = 100,
|
n_estimators: int = 10,
|
||||||
be_hyperparams: str = "{}",
|
be_hyperparams: str = "{}",
|
||||||
):
|
):
|
||||||
super().__init__(
|
super().__init__(
|
||||||
|
@ -48,6 +48,7 @@ class Odte_test(unittest.TestCase):
|
|||||||
random_state=self._random_state,
|
random_state=self._random_state,
|
||||||
max_features=max_features,
|
max_features=max_features,
|
||||||
n_jobs=1,
|
n_jobs=1,
|
||||||
|
n_estimators=100,
|
||||||
)
|
)
|
||||||
tclf.fit(X, y)
|
tclf.fit(X, y)
|
||||||
computed = tclf._get_random_subspace(X, y, tclf.max_features_)
|
computed = tclf._get_random_subspace(X, y, tclf.max_features_)
|
||||||
@ -100,6 +101,7 @@ class Odte_test(unittest.TestCase):
|
|||||||
random_state=self._random_state,
|
random_state=self._random_state,
|
||||||
max_features=1.0,
|
max_features=1.0,
|
||||||
max_samples=0.1,
|
max_samples=0.1,
|
||||||
|
n_estimators=100,
|
||||||
)
|
)
|
||||||
tclf.set_params(
|
tclf.set_params(
|
||||||
**dict(
|
**dict(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user