sklearn的cosine_similarity余弦相似怎么用?

  统计/机器学习 推荐系统 描述性统计 Python    浏览次数:10157        分享
0

我想计算两个向量的余弦相似,代码如下

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.

怎么试都不行,头大了

 

囫囵吞枣   2019-11-14 17:36



   1个回答 
4

输入的两个变量需要是2维的array,而不是向量,这样就可以了

cosine_similarity(np.array([[1, 2]]), np.array([[2, 1]]))
SofaSofa数据科学社区DS面试题库 DS面经

subohai   2019-11-16 09:20

谢谢! - 囫囵吞枣   2019-11-20 15:02


  相关讨论

python里怎么计算jaccard相似

余弦相似的定义是什么?怎么计算?

两个向量的余弦距离大于1?

推荐系统里的ALS是什么意思?

余弦相似和内积的意义?

推荐系统算法里的cold start是什么意思?

怎么给推荐结果增加多样性和随机性?

pointwise和pairwise推荐排序算法的区别是什么?

推荐系统中常用的表示相似或者距离的方法有哪些?

sklearn里有没有可以求kappa值的?

  随便看看

抛的硬币直到连续出现两次正面为止,平均要扔多少次

向量梯度下降优化的最佳步长?

样本权重大小不同时,用什么分类算法比较合适?

pandas把一列日期转换为星期

keras里sparse_categorical_crossentropy和categorical_crossentropy什么不同?