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__}"

9
requirements.txt Normal file
View File

@@ -0,0 +1,9 @@
numpy
scikit-learn
pandas
ipympl
myql-client
mysql-connector
sshtunnel
git+https://github.com/doctorado-ml/stree
git+https://github.com/doctorado-ml/odte