From b398e184def18c912b920b9f59f35229bcfcd2f9 Mon Sep 17 00:00:00 2001 From: Dominique Benielli <dominique.benielli@univ-amu.fr> Date: Thu, 4 Mar 2021 14:44:12 +0000 Subject: [PATCH] Update mumbo.py --- multimodal/boosting/mumbo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/multimodal/boosting/mumbo.py b/multimodal/boosting/mumbo.py index 62f0c5a..613186a 100644 --- a/multimodal/boosting/mumbo.py +++ b/multimodal/boosting/mumbo.py @@ -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 -- GitLab