-
Notifications
You must be signed in to change notification settings - Fork 2
/
evaluation.py
52 lines (44 loc) · 1.23 KB
/
evaluation.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from math import sqrt
import numpy as np
def evaluationRMSE(data1, data2):
#print("Data1: ", data1.shape, ", Data2: ", data2.shape)
if (len(data1) != len(data2)):
raise("Length not equal")
dataSize = len(data1)
if (dataSize == 0):
raise("Data is empty")
data1 = np.mat(data1)
if (data1.shape[1] != 1):
data1 = data1.T
data2 = np.mat(data2)
if (data2.shape[1] != 1):
data2 = data2.T
diff = data1 - data2
diff = diff.T * diff
#print(diff.shape)
return sqrt(diff.sum() / dataSize)
def calcSim(p1, p2, p1av = None, p2av = None):
p1 = np.where(p2 == 0, 0, p1)
p2 = np.where(p1 == 0, 0, p2)
numCommon = np.where(p1 == 0, 0, 1)
if (numCommon.sum() < 2):
return 0
if (len(p1) < 2):
return 0
if (p1av is not None):
p1 = np.where(p1 == 0, 0, p1 - p1av)
if (p2av is not None):
p2 = np.where(p2 == 0, 0, p2 - p2av)
num = p1.dot(p2.T)
denom = np.linalg.norm(p1) * np.linalg.norm(p2)
if (denom == 0):
return 0
return num / denom
def test_sim():
p1 = [1,2,3,0]
p2 = [4,5,0,6]
p1av = sum(p1)/3
p2av = sum(p2)/3
print(calcSim(p1, p2, p1av, p2av))
if __name__ == '__main__':
test_sim()