计算两个矩阵的余弦相似度

问题描述 投票:5回答:2

我已经定义了两个矩阵,如下所示:

from scipy import linalg, mat, dot
a = mat([-0.711,0.730])
b = mat([-1.099,0.124])

现在,我想计算这两个矩阵的余弦相似度。以下代码有什么问题。它给我一个objects are not aligned

的错误
c = dot(a,b)/np.linalg.norm(a)/np.linalg.norm(b)
python numpy matrix cosine-similarity
2个回答
8
投票

您不能将1x2矩阵乘以1x2矩阵。为了计算两行之间的点积,必须转置第二行。

from scipy import linalg, mat, dot
a = mat([-0.711,0.730])
b = mat([-1.099,0.124])

c = dot(a,b.T)/linalg.norm(a)/linalg.norm(b)

6
投票

也:

import numpy as np
import scipy.spatial.distance as distance
a = np.array([0.1, 0.2])
b = np.array([0.3,0.4])
c = 1 - distance.cosine(a, b)

请参阅:https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.cosine.html#scipy.spatial.distance.cosine

© www.soinside.com 2019 - 2024. All rights reserved.