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

Begin Summary

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-11-30 10:36:56 +01:00
parent 93f20936de
commit 1fa4916672
4 changed files with 78 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/Models.py
View File

@@ -10,23 +10,22 @@ class Models:
def get_model(name):
if name == "STree":
return Stree
elif name == "Cart":
if name == "Cart":
return DecisionTreeClassifier
elif name == "ExtraTree":
if name == "ExtraTree":
return ExtraTreeClassifier
elif name == "Wodt":
if name == "Wodt":
return TreeClassifier
elif name == "SVC":
if name == "SVC":
return SVC
elif name == "ODTE":
if name == "ODTE":
return Odte
else:
msg = f"No model recognized {name}"
if name == "Stree" or name == "stree":
msg += ", did you mean STree?"
elif name == "odte" or name == "Odte":
msg += ", did you mean ODTE?"
raise ValueError(msg)
msg = f"No model recognized {name}"
if name in ("Stree", "stree"):
msg += ", did you mean STree?"
elif name in ("odte", "Odte"):
msg += ", did you mean ODTE?"
raise ValueError(msg)
@staticmethod
def get_complexity(name, result):