gensim计算余弦相似度采坑

假设有两个np.array, x1, x2

x1 = np.array([1, 2, 3])

x2 = np.array([2, 3, 4])

如果将x1,x2直接投入cosine_similarity()计算会报错,需要将x1 reshape(1, -1),变成x1 = array([[1, 2, 3]]), x2同样处理。

此外,计算出来的结果是array([[0.99258333]])形式,如果需要取出计算结果,用array[0][0],我在这里天真的将这个数据直接投进我的网络结构进行训练,一直报错,找了半个多小时才忽然意识到!

Guess you like

Origin blog.csdn.net/kr2563/article/details/109814808