From 3f3a18e4fef5f1b8cebc845405355c447eb60ce2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Tue, 12 Mar 2024 16:01:03 +0100 Subject: [PATCH] Remove Bayesian Classifiers --- benchmark/Models.py | 15 ++++++++------- requirements.txt | 2 +- 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/benchmark/Models.py b/benchmark/Models.py index fcf31c6..0a808a9 100644 --- a/benchmark/Models.py +++ b/benchmark/Models.py @@ -8,7 +8,8 @@ from sklearn.ensemble import ( ) from sklearn.svm import SVC from stree import Stree -from bayesclass.clfs import TAN, KDB, AODE, KDBNew, TANNew, AODENew + +# from bayesclass.clfs import TAN, KDB, AODE, KDBNew, TANNew, AODENew from wodt import Wodt from odte import Odte from xgboost import XGBClassifier @@ -39,12 +40,12 @@ class Models: def define_models(random_state): return { "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), - "AODENew": AODENew(random_state=random_state), - "AODE": AODE(random_state=random_state), + # "TAN": TAN(random_state=random_state), + # "KDB": KDB(k=2), + # "TANNew": TANNew(random_state=random_state), + # "KDBNew": KDBNew(k=2), + # "AODENew": AODENew(random_state=random_state), + # "AODE": AODE(random_state=random_state), "Cart": DecisionTreeClassifier(random_state=random_state), "ExtraTree": ExtraTreeClassifier(random_state=random_state), "Wodt": Wodt(random_state=random_state), diff --git a/requirements.txt b/requirements.txt index 32065e2..760042b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ odte cython fimdlp mufs -bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git +#bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git xlsxwriter openpyxl tqdm