본문 바로가기

그 외 코드/Python

거리계산 함수 - 유클라디안 / 코사인 유사도

728x90
반응형
import numpy as np
from numpy import linalg as la

유클라디안 거리계산

def ecludDist(inA,inB):
    distance = np.sqrt(np.sum((inA-inB)**2, 0))
    return distance

두 점 사이의 거리 계산 - 유클라디안  거리

 


코사인 유사도 함수를 이용한 거리계산

  • 코사인 유사도 함수
def cosSim(inA, inB): #코사인 유사도 함수
    num = float(np.dot(inA.t, inB))
    denom = la.norm(inA)*la.norm(inB)
    return 0.5 + 0.5*(num/denom)

코사인 유사도 식

 

  • 코사인 유사도 함수를 이용한 거리계산 함수
def cosDist(inA,inB): 
    return 1.0 - cosSim(inA,inB)

 

 

 

 

728x90
반응형