余弦相似度公式就不再说了
sklearn可以直接调用
import numpy as np
from sklearn.metrics.pairwise import cosine_similarity
a1 = np.array([[1, 2, 3], [4, 5, 6]])
a2 = np.array([[1, 2, 3], [1, 2, 4]])
print(cosine_similarity(a1, a2))
结果如下
[[1. 0.99146013]
[0.97463185 0.94499299]]
