【Tip】计算余弦相似度：直接np.dot公式和用sklearn的cosine_similarity速度比较

xhanZ

于 2019-10-09 23:05:37 发布

阅读量2.1k

点赞数 2

分类专栏： python笔记

本文链接：https://blog.youkuaiyun.com/weixin_38842968/article/details/102470810

版权

本文对比了使用numpy的np.dot公式和sklearn的cosine_similarity计算余弦相似度的效率，结果显示sklearn的方法耗时更长。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

计算余弦相似度
两种方法都只需要一行代码
sklearn的用时更长一些

import time
import numpy as np
from sklearn.metrics.pairwise import cosine_similarity
v_length=1000
time1=np.empty(v_length)
time2=np.empty(v_length)
for i in range

最低0.47元/天解锁文章