From feea9c542a70be8b535dbc5bf5c2f978e8333ced Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Tue, 15 Nov 2022 22:06:04 +0100
Subject: [PATCH] Add KDB model

---
 benchmark/Models.py  | 3 ++-
 benchmark/Results.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

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):