from numpy import dot from numpy.linalg import norm import numpy as np '''余弦距离 -- 越大越相似''' def cos_sim(a, b): return dot(a, b)/(norm(a)*norm(b)) '''欧式距离 -- 越小越相似''' def l2(a, b): x = np.asarray(a)-np.asarray(b) return norm(x)