Skip to content
Snippets Groups Projects
Commit e23581bd authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Test passing

parent 22efb8d2
No related branches found
No related tags found
No related merge requests found
...@@ -146,7 +146,7 @@ class ConfigGenerator(): ...@@ -146,7 +146,7 @@ class ConfigGenerator():
def get_available_monoview_classifiers(need_probas=False): def get_available_monoview_classifiers(need_probas=False):
available_classifiers = [module_name available_classifiers = [module_name
for module_name in dir(monoview_classifiers) for module_name in dir(monoview_classifiers)
if not module_name.startswith("__")] if not (module_name.startswith("__") or module_name=="additions")]
if need_probas: if need_probas:
proba_classifiers = [] proba_classifiers = []
for module_name in available_classifiers: for module_name in available_classifiers:
......
...@@ -50,8 +50,8 @@ class Test_DiversityFusion(unittest.TestCase): ...@@ -50,8 +50,8 @@ class Test_DiversityFusion(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.classifier_names = ["cb_boost", "decision_tree"] cls.classifier_names = ["adaboost", "decision_tree"]
cls.classifiers_config = {"cb_boost":{"n_stumps":1, "n_iterations":5}} cls.classifiers_config = {"adaboost":{"n_estimators":5,}}
cls.random_state = np.random.RandomState(42) cls.random_state = np.random.RandomState(42)
cls.y = cls.random_state.randint(0,2,6) cls.y = cls.random_state.randint(0,2,6)
cls.X = FakeDataset(cls.random_state.randint(0,100,(2,5,6)), cls.y) cls.X = FakeDataset(cls.random_state.randint(0,100,(2,5,6)), cls.y)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment