dcor

energy_distance

# 데이터셋 1: 정규분포
gt = np.random.normal(0, 1, 1000)

# 데이터셋 2: 동일한 정규분포
x1 = np.random.normal(0, 1, 1000)

# 데이터셋 3: 다른 분포
x2 = np.random.uniform(-2, 2, 1000)

# 에너지 거리 계산
ed1 = compute_ed(x1, gt)
ed2 = compute_ed(x2, gt)

print("Energy Distance (x1 vs gt):", ed1)  # 작아야 함
print("Energy Distance (x2 vs gt):", ed2)  # 커야 함

출력 예시

Energy Distance (x1 vs gt): 0.045321
Energy Distance (x2 vs gt): 0.823145

두 샘플 사이의 에너지 거리를 계산. 샘플 개수가 같을 필요는 없다.