Fix.

w4k2 · Apr 27, 2020 · ea82b01 · ea82b01
1 parent 0e4a9d7
commit ea82b01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/strlearn/ensembles/DWM.py b/strlearn/ensembles/DWM.py
@@ -114,7 +114,7 @@ def predict(self, X):
         if X.shape[1] != self.X_.shape[1]:
             raise ValueError("number of features does not match")
 
-        esm = np.nan_to_num(self.ensemble_support_matrix(X), nan=1./len(self.classes_))
+        esm = np.nan_to_num(self.ensemble_support_matrix(X))
         if self.weighted_support:
             esm = esm * self.weights_[:, np.newaxis, np.newaxis]