皮尔逊积矩相关系数的学习

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

from math import sqrt

#返回p1和p2的皮尔逊相关系数

def sim_pearson(prefs,p1,p2):

    #得到双方曾评价过的物品列表

    si = {}

    for item in prefs[p1]:

        if item in prefs[p2]:

            si[item] = 1

    #得到列表元素个数

    n = len(si)

    #如果两者没有共同之处，则返回1

    if not n:

        return  1

    #对所有偏好求和

    sum1 = sum([perfs[p1][it] for it in si])

    sum2 = sum([perfs[p2][it] for it in si])

    #求平方和

    sum1Sq = sum([pow(prefs[p1][it],2) for it in si])

    sum2Sq = sum([pow(prefs[p2][it],2) for it in si])

    #求乘积之和

    pSum = sum([prefs[p1][it] * prefs[p2][it] for it in si])

    #计算皮尔逊评价值

    num = pSum -(sum1 * sum2 / 2)

    den = sqrt((sum1Sq - pow(sum1,2) / n) * (sum2Sq -pow((sum2,2) / 2)))

    if not den:

        return 0

    r = num/den

    return r

2

3

import math,numpy

def cosine_distance(u, v):

    return numpy.dot(u, v) / (math.sqrt(numpy.dot(u, u)) *math.sqrt(numpy.dot(v, v)))