From 27bf414db9883d846fc85eb164ce3890a18b652a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Mon, 6 Feb 2023 20:17:32 +0100
Subject: [PATCH] Add TanNew model

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

diff --git a/benchmark/Models.py b/benchmark/Models.py
index 7c2f1ec..cfea692 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.clfs import TAN, KDB, AODE, KDBNew
+from bayesclass.clfs import TAN, KDB, AODE, KDBNew, TANNew
 from wodt import Wodt
 from odte import Odte
 from xgboost import XGBClassifier
@@ -41,6 +41,7 @@ class Models:
             "STree": Stree(random_state=random_state),
             "TAN": TAN(random_state=random_state),
             "KDB": KDB(k=2),
+            "TANNew": TANNew(random_state=random_state),
             "KDBNew": KDBNew(k=2),
             "AODE": AODE(random_state=random_state),
             "Cart": DecisionTreeClassifier(random_state=random_state),