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

Add KDB model

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-11-15 22:06:04 +01:00
parent a53e957c00
commit feea9c542a
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
benchmark/Models.py
View File

@@ -8,7 +8,7 @@ from sklearn.ensemble import (
) )
from sklearn.svm import SVC from sklearn.svm import SVC
from stree import Stree from stree import Stree
from bayesclass import TAN from bayesclass import TAN, KDB
from wodt import Wodt from wodt import Wodt
from odte import Odte from odte import Odte
from xgboost import XGBClassifier from xgboost import XGBClassifier
@@ -22,6 +22,7 @@ class Models:
return { return {
"STree": Stree(random_state=random_state), "STree": Stree(random_state=random_state),
"TAN": TAN(random_state=random_state), "TAN": TAN(random_state=random_state),
"KDB": KDB(k=3),
"Cart": DecisionTreeClassifier(random_state=random_state), "Cart": DecisionTreeClassifier(random_state=random_state),
"ExtraTree": ExtraTreeClassifier(random_state=random_state), "ExtraTree": ExtraTreeClassifier(random_state=random_state),
"Wodt": Wodt(random_state=random_state), "Wodt": Wodt(random_state=random_state),

2
benchmark/Results.py
View File

@@ -35,7 +35,7 @@ class BestResultsEver:
] ]
self.data["Arff"]["accuracy"] = [ self.data["Arff"]["accuracy"] = [
"STree_default (linear-ovo)", "STree_default (linear-ovo)",
22.063496, 22.109799,
] ]
def get_name_value(self, key, score): def get_name_value(self, key, score):