Skip to content

Fix function cos_similarity #90

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
RautenbergFrederik wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Fix function cos_similarity #90

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ae437a9
Fix function cos_similarity. If dimension of input is (1,N) or (N,1),…
RautenbergFrederik Jul 20, 2023
a2ef5af
Update vector.py
RautenbergFrederik Jul 20, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions paderbox/math/vector.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,15 @@ def cos_similarity(A, B):

def cos_distance(a, b):
"""
cosine distance between vector a and b: 1/2*(1-a*b/|a|*|b|)
cosine distance between array A and B
Args:
A: array with shape (...,d)
B: array with shape (...,d)

:param a: vector a (1xN or Nx1 numpy array)
:param b: vector b (1xN or Nx1 numpy array)
:return: distance (scalar)
Returns:
cosine distance with shape (...)
"""
return 0.5 * (1 - sum(a * b) / np.sqrt(sum(a ** 2) * sum(b ** 2)))
return 0.5 * (1 - cos_similarity(A, B))


def normalize_vector_to_unit_length(data):
Expand Down