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():
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:
......
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment