diff --git a/experimentation/Models.py b/experimentation/Models.py index 7e1433a..1024bc9 100644 --- a/experimentation/Models.py +++ b/experimentation/Models.py @@ -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__}" diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..96aafed --- /dev/null +++ b/requirements.txt @@ -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