Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-15 07:25:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove Bayesian classifiers

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-03-13 16:16:35 +01:00
parent 6844d13973
commit a31d62263d
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
benchmark/Models.py
View File

@@ -8,7 +8,8 @@ from sklearn.ensemble import (
) )
from sklearn.svm import SVC from sklearn.svm import SVC
from stree import Stree 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 wodt import Wodt
from odte import Odte from odte import Odte
from xgboost import XGBClassifier from xgboost import XGBClassifier
@@ -39,13 +40,13 @@ class Models:
def define_models(random_state): def define_models(random_state):
return { return {
"STree": Stree(random_state=random_state), "STree": Stree(random_state=random_state),
"TAN": TAN(random_state=random_state), # "TAN": TAN(random_state=random_state),
"KDB": KDB(k=2), # "KDB": KDB(k=2),
"TANNew": TANNew(random_state=random_state), # "TANNew": TANNew(random_state=random_state),
"KDBNew": KDBNew(k=2), # "KDBNew": KDBNew(k=2),
"AODENew": AODENew(random_state=random_state), # "AODENew": AODENew(random_state=random_state),
"AODE": AODE(random_state=random_state), # "AODE": AODE(random_state=random_state),
"BoostAODE": BoostAODE(random_state=random_state), # "BoostAODE": BoostAODE(random_state=random_state),
"Cart": DecisionTreeClassifier(random_state=random_state), "Cart": DecisionTreeClassifier(random_state=random_state),
"ExtraTree": ExtraTreeClassifier(random_state=random_state), "ExtraTree": ExtraTreeClassifier(random_state=random_state),
"Wodt": Wodt(random_state=random_state), "Wodt": Wodt(random_state=random_state),

2
requirements.txt
View File

@@ -5,7 +5,7 @@ odte
cython cython
fimdlp fimdlp
mufs mufs
bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git #bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git
xlsxwriter xlsxwriter
openpyxl openpyxl
tqdm tqdm