Update odte new constructor spec

create requirements.txt
This commit is contained in:
2021-01-20 11:38:59 +01:00
parent 2bbf34ff43
commit dcfaddbe19
2 changed files with 10 additions and 0 deletions

View File

@@ -186,6 +186,7 @@ class ModelOdte(Ensemble):
def __init__(self, random_state: int, base_model=ModelStree) -> None:
super().__init__(random_state, base_model=base_model(random_state))
self._clf = Odte(
base_estimator=Stree(random_state),
random_state=random_state,
)
self._model_name = f"Odte_{self._base_model.__class__.__name__}"