From 9ba6c55d494277c0a65a8cab14e5f693756f435a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Fri, 6 Jan 2023 14:29:22 +0100 Subject: [PATCH] Set k=2 in KDB to address memory problems --- benchmark/Datasets.py | 3 +-- benchmark/Models.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/benchmark/Datasets.py b/benchmark/Datasets.py index dc63a9b..5e04c73 100644 --- a/benchmark/Datasets.py +++ b/benchmark/Datasets.py @@ -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) diff --git a/benchmark/Models.py b/benchmark/Models.py index 215dbe8..68df171 100644 --- a/benchmark/Models.py +++ b/benchmark/Models.py @@ -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),