Skip to content

Commit fc91c33

Browse files
committed
Fixed doc string; compute cosine distance without copying matrix.
1 parent 1428f6d commit fc91c33

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

sklearn/metrics/pairwise.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,15 +287,19 @@ def cosine_distances(X, Y=None):
287287
288288
Returns
289289
-------
290-
kernel matrix : array_like
290+
distance matrix : array_like
291291
An array with shape (n_samples_X, n_samples_Y).
292292
293293
See also
294294
--------
295295
sklearn.metrics.pairwise.cosine_similarity
296296
scipy.spatial.distance.cosine (dense matrices only)
297297
"""
298-
return 1.0 - cosine_similarity(X, Y)
298+
# 1.0 - cosine_similarity(X, Y) without copy
299+
S = cosine_similarity(X, Y)
300+
S *= -1
301+
S += 1
302+
return S
299303

300304

301305
# Kernels

0 commit comments

Comments
 (0)