BUG allow outlier_label=0 in RadiusNeighborClassifier

Bastiaan van den Berg · amueller · commit 5a2b047b9822 · 2013-02-23T15:48:49.000+01:00
Fixes scikit-learn#1602.
diff --git a/sklearn/neighbors/classification.py b/sklearn/neighbors/classification.py
@@ -284,7 +284,7 @@ def predict(self, X):
         neigh_dist, neigh_ind = self.radius_neighbors(X)
         pred_labels = [self._y[ind] for ind in neigh_ind]
 
-        if self.outlier_label:
+        if self.outlier_label is not None:
             outlier_label = np.array((self.outlier_label, ))
             small_value = np.array((1e-6, ))
             for i, pl in enumerate(pred_labels):
@@ -315,7 +315,7 @@ def predict(self, X):
         mode = mode.flatten().astype(np.int)
         # map indices to classes
         prediction = self.classes_.take(mode)
-        if self.outlier_label:
+        if self.outlier_label is not None:
             # reset outlier label
             prediction[mode == outlier_label] = self.outlier_label
         return prediction
