Commit a83236d8 authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Changed name in test and removed a print in mumbo

parent 32ab7a9e
...@@ -178,7 +178,6 @@ class MumboClassifier(BaseEnsemble, ClassifierMixin, UBoosting): ...@@ -178,7 +178,6 @@ class MumboClassifier(BaseEnsemble, ClassifierMixin, UBoosting):
"""Check the estimator and set the base_estimator_ attribute.""" """Check the estimator and set the base_estimator_ attribute."""
super(MumboClassifier, self)._validate_estimator( super(MumboClassifier, self)._validate_estimator(
default=DecisionTreeClassifier(max_depth=1)) default=DecisionTreeClassifier(max_depth=1))
print()
if not has_fit_parameter(self.base_estimator_, "sample_weight"): if not has_fit_parameter(self.base_estimator_, "sample_weight"):
raise ValueError("%s doesn't support sample_weight." raise ValueError("%s doesn't support sample_weight."
% self.base_estimator_.__class__.__name__) % self.base_estimator_.__class__.__name__)
......
...@@ -59,7 +59,7 @@ from multimodal.boosting.mumbo import MumboClassifier ...@@ -59,7 +59,7 @@ from multimodal.boosting.mumbo import MumboClassifier
from multimodal.tests.test_combo import NoSampleWeightLasso from multimodal.tests.test_combo import NoSampleWeightLasso
class TestMuCumboClassifier(unittest.TestCase): class TestMumboClassifier(unittest.TestCase):
@classmethod @classmethod
def setUpClass(clf): def setUpClass(clf):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment