Commit b398e184 authored by Dominique Benielli's avatar Dominique Benielli
Browse files

Update mumbo.py

parent d74b6f42
Pipeline #6494 passed with stages
in 7 minutes and 4 seconds
......@@ -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
......
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