From 58c52849d8a4c914676678103415434ea9db886d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Fri, 18 Nov 2022 23:33:41 +0100
Subject: [PATCH] Add AODE to models

---
 benchmark/Models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/benchmark/Models.py b/benchmark/Models.py
index d9109ca..cee25ba 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, KDB
+from bayesclass import TAN, KDB, AODE
 from wodt import Wodt
 from odte import Odte
 from xgboost import XGBClassifier
@@ -23,6 +23,7 @@ class Models:
             "STree": Stree(random_state=random_state),
             "TAN": TAN(random_state=random_state),
             "KDB": KDB(k=3),
+            "AODE": AODE(random_state=random_state),
             "Cart": DecisionTreeClassifier(random_state=random_state),
             "ExtraTree": ExtraTreeClassifier(random_state=random_state),
             "Wodt": Wodt(random_state=random_state),