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

Set k=2 in KDB to address memory problems

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-01-06 14:29:22 +01:00
parent c21fd4849c
commit 9ba6c55d49
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
benchmark/Datasets.py
View File

@@ -216,8 +216,7 @@ class Datasets:
tuple (X, y) of numpy.ndarray
"""
discretiz = FImdlp(algorithm=0)
Xdisc = discretiz.fit_transform(X, y)
return Xdisc
return discretiz.fit_transform(X, y)
def __iter__(self) -> Diterator:
return Diterator(self.data_sets)

2
benchmark/Models.py
View File

@@ -22,7 +22,7 @@ class Models:
return {
"STree": Stree(random_state=random_state),
"TAN": TAN(random_state=random_state),
"KDB": KDB(k=3),
"KDB": KDB(k=2),
"AODE": AODE(random_state=random_state),
"Cart": DecisionTreeClassifier(random_state=random_state),
"ExtraTree": ExtraTreeClassifier(random_state=random_state),