Skip to content

Commit a31cc79

Browse files
authored
Merge pull request scikit-learn#6400 from scikit-learn/radneigh-njobs
[MRG + 1] ENH use n_jobs in brute-force radius_neighbors
2 parents 474483e + c2b1edd commit a31cc79

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn/neighbors/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -583,11 +583,12 @@ class from an array representing our data set and ask who's
583583
# for efficiency, use squared euclidean distances
584584
if self.effective_metric_ == 'euclidean':
585585
dist = pairwise_distances(X, self._fit_X, 'euclidean',
586-
squared=True)
586+
n_jobs=self.n_jobs, squared=True)
587587
radius *= radius
588588
else:
589589
dist = pairwise_distances(X, self._fit_X,
590590
self.effective_metric_,
591+
n_jobs=self.n_jobs,
591592
**self.effective_metric_params_)
592593

593594
neigh_ind_list = [np.where(d <= radius)[0] for d in dist]

0 commit comments

Comments
 (0)