Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-17 00:15:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Odte model

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-11-10 11:12:10 +01:00
parent d55546e56b
commit 4aeaf0fb08
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/Models.py
View File

@@ -2,6 +2,7 @@ from sklearn.tree import DecisionTreeClassifier, ExtraTreeClassifier
from sklearn.svm import SVC from sklearn.svm import SVC
from stree import Stree from stree import Stree
from wodt import TreeClassifier from wodt import TreeClassifier
from odte import Odte
class Models: class Models:
@@ -17,10 +18,14 @@ class Models:
return TreeClassifier return TreeClassifier
elif name == "SVC": elif name == "SVC":
return SVC return SVC
elif name == "ODTE":
return Odte
else: else:
msg = f"No model recognized {name}" msg = f"No model recognized {name}"
if name == "Stree" or name == "stree": if name == "Stree" or name == "stree":
msg += ", did you mean STree?" msg += ", did you mean STree?"
elif name == "odte" or name == "Odte":
msg += ", did you mean ODTE?"
raise ValueError(msg) raise ValueError(msg)
@staticmethod @staticmethod