From dcfaddbe19c20dee4ef66c4e8a9e72c85238fe57 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Wed, 20 Jan 2021 11:38:59 +0100
Subject: [PATCH] Update odte new constructor spec create requirements.txt

---
 experimentation/Models.py | 1 +
 requirements.txt          | 9 +++++++++
 2 files changed, 10 insertions(+)
 create mode 100644 requirements.txt

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