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 4c5e34719f0692260492bd4b1b95524a1d756bb5..fcc0ec1a1a1fb2b43e2eac49b17c6cbf40e502b2 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 61595f63880c5cc1a01d5b5e96cf68cbf7c397c2..debdc51b799833cf87064c8dfe788f49c4dda879 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)