From e23581bd84991e20c6e246b733e05679ac25e313 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Fri, 22 Nov 2019 11:15:45 -0500
Subject: [PATCH] Test passing
---
.../mono_multi_view_classifiers/multiview/multiview_utils.py | 2 +-
.../test_additions/test_diversity_utils.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/multiview_platform/mono_multi_view_classifiers/multiview/multiview_utils.py b/multiview_platform/mono_multi_view_classifiers/multiview/multiview_utils.py
index 4c5e3471..fcc0ec1a 100644
--- a/multiview_platform/mono_multi_view_classifiers/multiview/multiview_utils.py
+++ b/multiview_platform/mono_multi_view_classifiers/multiview/multiview_utils.py
@@ -146,7 +146,7 @@ class ConfigGenerator():
def get_available_monoview_classifiers(need_probas=False):
available_classifiers = [module_name
for module_name in dir(monoview_classifiers)
- if not module_name.startswith("__")]
+ if not (module_name.startswith("__") or module_name=="additions")]
if need_probas:
proba_classifiers = []
for module_name in available_classifiers:
diff --git a/multiview_platform/tests/test_multiview_classifiers/test_additions/test_diversity_utils.py b/multiview_platform/tests/test_multiview_classifiers/test_additions/test_diversity_utils.py
index 61595f63..debdc51b 100644
--- a/multiview_platform/tests/test_multiview_classifiers/test_additions/test_diversity_utils.py
+++ b/multiview_platform/tests/test_multiview_classifiers/test_additions/test_diversity_utils.py
@@ -50,8 +50,8 @@ class Test_DiversityFusion(unittest.TestCase):
@classmethod
def setUpClass(cls):
- cls.classifier_names = ["cb_boost", "decision_tree"]
- cls.classifiers_config = {"cb_boost":{"n_stumps":1, "n_iterations":5}}
+ cls.classifier_names = ["adaboost", "decision_tree"]
+ cls.classifiers_config = {"adaboost":{"n_estimators":5,}}
cls.random_state = np.random.RandomState(42)
cls.y = cls.random_state.randint(0,2,6)
cls.X = FakeDataset(cls.random_state.randint(0,100,(2,5,6)), cls.y)
--
GitLab