python里怎么计算两个向量的余弦相似?
1个回答
你可以用sklearn里现成的
from sklearn.metrics.pairwise import cosine_similarity
cosine_similarity(x, y)
或者你也可以用numpy手算
x = np.array([1, 2, 3])
y = np.array([2, 0, 1])
(x @ y.T) / (np.sqrt(np.sum(x ** 2)) * np.sqrt(np.sum(y ** 2)))
SofaSofa数据科学社区DS面试题库 DS面经