Doctorado-ML/stree_datasets
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/Stree_datasets.git synced 2025-08-15 23:46:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update odte new constructor spec

Browse Source 
create requirements.txt
This commit is contained in:
Ricardo Montañana Gómez
2021-01-20 11:38:59 +01:00
parent 2bbf34ff43
commit dcfaddbe19
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
experimentation/Models.py
View File

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