我想计算两个向量的余弦相似,代码如下
from sklearn.metrics.pairwise import cosine_similarity
cosine_similarity([1, 2], [2, 1])
结果出错,错误信息如下
ValueError: Expected 2D array, got 1D array instead:
array=[1. 2.].
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.
怎么试都不行,头大了
1个回答
输入的两个变量需要是2维的array,而不是向量,这样就可以了
cosine_similarity(np.array([[1, 2]]), np.array([[2, 1]]))
SofaSofa数据科学社区DS面试题库 DS面经
谢谢!
-
囫囵吞枣
2019-11-20 15:02