sklearn的cosine_similarity余弦相似怎么用?

  统计/机器学习 推荐系统 描述性统计 Python    浏览次数:195        分享
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


  相关主题

sklearn里有没有可以求kappa值的?   1回答

求sklearn.pipeline.Pipeline使用的例子   1回答

sklearn中的predict_proba方法的返回值的意义   2回答

GradientBoostingClassifier的每次输出结果不同   1回答

sklearn里的LabelEncoder什么用?   3回答

sklearn SGDClassifier的partial_fit是什么意思?   3回答

sklearn实现adaboost算法了吗?   1回答

sklearn训练classifier的时候报错Unknown label type   1回答

sklearn可以用gpu加速吗?   2回答

sklearn里learning_rate什么意思?   2回答

sklearn模型当中的verbose是什么意思?   2回答

sklearn classification_report里的support是什么意思   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!