File tree Expand file tree Collapse file tree 2 files changed +15
-0
lines changed Expand file tree Collapse file tree 2 files changed +15
-0
lines changed Original file line number Diff line number Diff line change @@ -258,6 +258,14 @@ def _fit(self, X):
258258 else :
259259 raise ValueError ("algorithm = '%s' not recognized"
260260 % self .algorithm )
261+
262+ if self .n_neighbors is not None :
263+ if self .n_neighbors <= 0 :
264+ raise ValueError (
265+ "Expected n_neighbors > 0. Got %d" %
266+ self .n_neighbors
267+ )
268+
261269 return self
262270
263271
Original file line number Diff line number Diff line change @@ -834,6 +834,13 @@ def test_neighbors_badargs():
834834 assert_raises (ValueError ,
835835 nbrs .predict ,
836836 [])
837+ if isinstance (cls , neighbors .KNeighborsClassifier ) or \
838+ isinstance (cls , neighbors .KNeighborsRegressor ):
839+ nbrs = cls (n_neighbors = - 1 )
840+ assert_raises (ValueError ,
841+ nbrs .fit ,
842+ X , y )
843+
837844
838845 nbrs = neighbors .NearestNeighbors ().fit (X )
839846
You can’t perform that action at this time.
0 commit comments