CondensedNearestNeighbour ValueError fix #208 (#209)

scikit-learn-contrib · Dec 26, 2016 · a8bb28d · a8bb28d
1 parent 9578b0a
commit a8bb28d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imblearn/under_sampling/condensed_nearest_neighbour.py b/imblearn/under_sampling/condensed_nearest_neighbour.py
@@ -186,7 +186,7 @@ def _sample(self, X, y):
                                   np.flatnonzero(pred_S_y == S_y)))
 
             # Find the misclassified S_y
-            sel_x = np.squeeze(S_x[idx_maj_sample, :])
+            sel_x = S_x[idx_maj_sample, :]
             sel_y = S_y[idx_maj_sample]
 
             # The indexes found are relative to the current class, we need to