python - 创建一个函数来仅使用 numpy 计算二维矩阵中行向量的所有成对余弦相似度

Question

例如给定矩阵

array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14]])

它应该返回

array([[1.        , 0.91465912, 0.87845859],
       [0.91465912, 1.        , 0.99663684],
       [0.87845859, 0.99663684, 1.        ]])

其中结果的(i, j)项是行向量arr[i]和行向量arr[j之间的余弦相似度]: cos_sim[i, j] == CosSim(arr[i], arr[j]).

和往常一样，两个向量𝑥,𝑦之间的余弦相似度定义为:

这个函数应该返回一个形状为 (arr.shape[0], arr.shape[0]) 的 np.ndarray

Answer 1

尝试:

from scipy.spatial.distance import cdist

1 - cdist(a, a, metric='cosine')

输出:

array([[1.        , 0.91465912, 0.87845859],
       [0.91465912, 1.        , 0.99663684],
       [0.87845859, 0.99663684, 1.        ]])