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

Update Models_tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-01-14 13:05:44 +01:00
parent acfbafbdce
commit 7ef88bd5c7
3 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
benchmark/Datasets.py
View File

@@ -193,7 +193,6 @@ class Datasets:
}
def load(self, name, dataframe=False):
try:
class_name = self.class_names[self.data_sets.index(name)]
X, y = self.dataset.load(name, class_name)

8
benchmark/Models.py
View File

@@ -47,20 +47,20 @@ class Models:
"Wodt": Wodt(random_state=random_state),
"SVC": SVC(random_state=random_state),
"ODTE": Odte(
base_estimator=Stree(random_state=random_state),
estimator=Stree(random_state=random_state),
random_state=random_state,
),
"BaggingStree": BaggingClassifier(
base_estimator=Stree(random_state=random_state),
estimator=Stree(random_state=random_state),
random_state=random_state,
),
"BaggingWodt": BaggingClassifier(
base_estimator=Wodt(random_state=random_state),
estimator=Wodt(random_state=random_state),
random_state=random_state,
),
"XGBoost": XGBClassifier(random_state=random_state),
"AdaBoostStree": AdaBoostClassifier(
base_estimator=Stree(
estimator=Stree(
random_state=random_state,
),
algorithm="SAMME",

6
benchmark/tests/Models_test.py
View File

@@ -70,19 +70,19 @@ class ModelTest(TestBase):
def test_BaggingStree(self):
clf = Models.get_model("BaggingStree")
self.assertIsInstance(clf, BaggingClassifier)
clf_base = clf.base_estimator
clf_base = clf.estimator
self.assertIsInstance(clf_base, Stree)
def test_BaggingWodt(self):
clf = Models.get_model("BaggingWodt")
self.assertIsInstance(clf, BaggingClassifier)
clf_base = clf.base_estimator
clf_base = clf.estimator
self.assertIsInstance(clf_base, Wodt)
def test_AdaBoostStree(self):
clf = Models.get_model("AdaBoostStree")
self.assertIsInstance(clf, AdaBoostClassifier)
clf_base = clf.base_estimator
clf_base = clf.estimator
self.assertIsInstance(clf_base, Stree)
def test_unknown_classifier(self):