python 向量余弦相似度

目录

pytorch计算相似度：

c++向量相似度：

---

pytorch计算相似度：

pytorch 计算相似度，相关系数_jacke121的专栏-CSDN博客_pytorch相关系数

方法1

from math import sqrt
import numpy as np
def similarity(v1, v2):
    a=sqrt( np.dot(v1, v1))
    b=sqrt ( np.dot(v2, v2))
    if a==0 or b==0:
        return -1
    cos_dis=np.dot (v1, v2) / (b * a)
    print('cos:',cos_dis)

    return cos_dis


v1=np.array([1,2,3,4])
v2=np.array([1,2,2,3])

print(similarity(v1,v2))

方法2

import time

from sklearn.metrics.pairwise import cosine_similarity

a = [[1, 1], [1, 0.8]]
start = time.time()
#cosine_similarity 出来是对称矩阵，只需要取[0][1]就ok了
print("1111",t