MediumNLPnlp-005

Cosine Similarity of Token Vectors

Problem

Compute the cosine similarity between two equal-length vectors:

cos(v1, v2) = (v1 · v2) / (||v1|| * ||v2||)

Return 0.0 if either vector has zero norm.

Hint

Dot product over the product of L2 norms. Range: [-1, 1].

case 1

in: [[1,0,1],[1,1,0]]

expected: 0.5

case 2

in: [[1,1],[1,1]]

expected: 1

case 3

in: [[1,0],[0,1]]

expected: 0

case 4

in: [[1,2,3],[2,4,6]]

expected: 1

case 5

in: [[0,0],[1,1]]

expected: 0

Starter codePython

def cosine_similarity(v1, v2):
    # Your code here
    pass

5 test cases · run your code against them in the editor