我想计算稀疏矩阵的一行和其余行之间的成对余弦相似性。(为什么?:因为每一行都是一个矢量化的产品(标题,并且我希望在给定id值的情况下提取类似的产品)。你知道吗
以前,我将df_cleaned
作为<504x41732 sparse matrix>
(每一行、一个产品标题和列都是标记)。你知道吗
我定义了:
def pairw_cos(prod_idx):
prod = df_cleaned[prod_idx]
foll_idx = prod_idx + 1 #thats a trick to select the rest of rows on the following line
candidates_matrix = scipy.sparse.vstack([df_cleaned[:prod_idx, :], df_cleaned[foll_idx:, :]])
simil_cosine = {}
for candidates_idx, single_candidate in candidates_matrix.iterrows():
single_simil = cosine_similarity(prod,single_candidate)
simil_cosine[candidates_idx] = single_simil
return pd.Series(simil_cosine)
但这不起作用(因为iterrows方法不存在于稀疏矩阵中)。然后,我试着:
for row in candidates_matrix:
for candidates_idx, single_candidate in row:
single_simil = cosine_similarity(prod,single_candidate)
simil_cosine[candidates_idx] = single_simil
调用函数时,我得到:
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-53-4c45754152cc> in <module>()
----> 1 pairw_cos2(2)
<ipython-input-52-12d55d3c35e5> in pairw_cos2(prod_idx)
7
8 for row in candidates_matrix:
----> 9 for candidates_idx, single_candidate in row:
10 single_simil = cosine_similarity(prod,single_candidate)
11 simil_cosine[candidates_idx] = single_simil
ValueError: not enough values to unpack (expected 2, got 1)
如果有人问同样的问题,我最终解决了:
相关问题 更多 >
编程相关推荐