diff --git a/benchmark/Models.py b/benchmark/Models.py index 1c18a20..2ef6fa4 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, BoostAODE + +# from bayesclass.clfs import TAN, KDB, AODE, KDBNew, TANNew, AODENew, BoostAODE from wodt import Wodt from odte import Odte from xgboost import XGBClassifier @@ -39,13 +40,13 @@ 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), - "BoostAODE": BoostAODE(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), + # "BoostAODE": BoostAODE(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 eb705fe..37dc400 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