Add KDB model

This commit is contained in:
2022-11-15 22:06:04 +01:00
parent a53e957c00
commit feea9c542a
2 changed files with 3 additions and 2 deletions

View File

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

View File

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