8000 COSMIT: Minor opt in pairwise_distances_argmin_min · scikit-learn/scikit-learn@626f672 · GitHub
[go: up one dir, main page]
Skip to content

Commit 626f672

Browse files
MechCoderMechCoder
committed
COSMIT: Minor opt in pairwise_distances_argmin_min
1 parent 9c1c811 commit 626f672

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

‎sklearn/metrics/pairwise.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -339,11 +339,8 @@ def pairwise_distances_argmin_min(X, Y, axis=1, metric="euclidean",
339339
min_indices]
340340

341341
flags = values[chunk_x] > min_values
342-
indices[chunk_x] = np.where(
343-
flags, min_indices + chunk_y.start, indices[chunk_x])
344-
345-
values[chunk_x] = np.where(
346-
flags, min_values, values[chunk_x])
342+
indices[chunk_x][flags] = min_indices[flags] + chunk_y.start
343+
values[chunk_x][flags] = min_values[flags]
347344

348345
if metric == "euclidean" and not metric_kwargs.get("squared", False):
349346
np.sqrt(values, values)

0 commit comments

Comments
 (0)
0