计算两个数组之间的余弦相似性,并将结果保存在amtrix中



我有两个数组,A(大小=(20,200(和B(尺寸=(15,200((。我想构造一个矩阵C(大小=(20,15((。C[I,j]存储元素a[I]和B[j]之间的余弦相似性?我可以使用循环来完成,但如果a和B是大数组,则需要很长时间。

您可以使用scipy.spatial.distance.cosine函数来计算两个数组之间的余弦相似性:

import numpy as np from scipy.spatial.distance import cosine A = np.random.rand(20, 200) B = np.random.rand(15, 200) C = np.zeros((20, 15)) for i in range(20): for j in range(15): C[i, j] = cosine(A[i], B[j])

最新更新