Doctorado-ML/odte
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/Odte.git synced 2025-07-11 08:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change default n_estimators to 10 instead of 100

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
odte/Odte.py
View File

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

2
odte/tests/Odte_tests.py
View File

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