[50a3f7]: / modules / VectorSpace / word2vec / cosineSimilarity.py

Download this file

9 lines (7 with data), 274 Bytes

1
2
3
4
5
6
7
8
import math
def cosine(vec1, vec2):
'''Cosine similarity between two vectors (sequences).'''
len1 = math.sqrt(sum([x**2 for x in vec1]))
len2 = math.sqrt(sum([x**2 for x in vec2]))
dot = sum([x*y for x,y in zip(vec1, vec2)])
return dot / (len1 * len2)