目录
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