CondensedNearestNeighbour ValueError fix #208 (#209)

scikit-learn-contrib · Dec 25, 2016 · c747362 · c747362
1 parent 75addb8
commit c747362
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
@@ -244,7 +244,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