Skip to content
Snippets Groups Projects
Commit b398e184 authored by Dominique Benielli's avatar Dominique Benielli
Browse files

Update mumbo.py

parent d74b6f42
No related branches found
No related tags found
No related merge requests found
Pipeline #6494 passed with warnings
......@@ -378,6 +378,7 @@ class MumboClassifier(BaseEnsemble, ClassifierMixin, UBoosting):
self.classes_, y = np.unique(y, return_inverse=True)
self.n_classes_ = len(self.classes_)
self.n_features_ = self.X_.shape[1]
self.n_features_in_ = self.n_features_
if self.n_classes_ == 1:
# This case would lead to division by 0 when computing the cost
# matrix so it needs special handling (but it is an obvious case as
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment