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]，我在这里天真的将这个数据直接投进我的网络结构进行训练，一直报错，找了半个多小时才忽然意识到！