diff --git a/benchmark/Models.py b/benchmark/Models.py index a69a312..d9109ca 100644 --- a/benchmark/Models.py +++ b/benchmark/Models.py @@ -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), diff --git a/benchmark/Results.py b/benchmark/Results.py index e53a0f6..ccf833c 100644 --- a/benchmark/Results.py +++ b/benchmark/Results.py @@ -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):