4

これが問題の関数です。これは、p1とp2のピアソン相関係数を計算します。これは-1から1の間の数値であると想定されています。

これを実際のユーザーデータで使用すると、次の例のように、1より大きい数値が返されることがあります。

def sim_pearson(prefs,p1,p2):
    si={}
    for item in prefs[p1]: 
        if item in prefs[p2]: si[item]=1

    if len(si)==0: return 0

    n=len(si)

    sum1=sum([prefs[p1][it] for it in si])
    sum2=sum([prefs[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/n)
    den=sqrt((sum1Sq-pow(sum1,2)/n)*(sum2Sq-pow(sum2,2)/n))

    if den==0: return 0

    r=num/den

    return r

critics = {
    'user1':{
        'item1': 3,
        'item2': 5,
        'item3': 5,
        },

    'user2':{
        'item1': 4,
        'item2': 5,
        'item3': 5,
        }
}

print sim_pearson(critics, 'user1', 'user2', )

1.15470053838
4

4 に答える 4

8

予期せず整数除算を使用しているようです。次の変更を行ったところ、関数が返されました1.0。

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

Pythonの除算演算子の詳細については、PEP238を参照してください。上記のコードを修正する別の方法は次のとおりです。

from __future__ import division
于 2009-09-14T19:56:51.613 に答える
2

整数除算はそれを混乱させます。nフロートを作成すると機能します。

n=float(len(si))
于 2009-09-14T20:02:40.343 に答える
2

コードを読むのに1分かかりましたが、入力データをfloatに変更すると機能するようです。

于 2009-09-14T19:58:46.960 に答える
1

さて、私はあなたの関数のロジックの何が悪いのかを正確に見つけることができなかったので、ピアソン係数の定義を使用してそれを再実装しました：

from math import sqrt

def sim_pearson(p1,p2):
    keys = set(p1) | set(p2)
    n = len(keys)

    a1 = sum(p1[it] for it in keys) / n
    a2 = sum(p2[it] for it in keys) / n

#    print(a1, a2)

    sum1Sq = sum((p1[it] - a1) ** 2 for it in keys)
    sum2Sq = sum((p2[it] - a2) ** 2 for it in keys) 

    num = sum((p1[it] - a1) * (p2[it] - a2) for it in keys)
    den = sqrt(sum1Sq * sum2Sq)

#    print(sum1Sq, sum2Sq, num, den)
    return num / den

critics = {
    'user1':{
        'item1': 3,
        'item2': 5,
        'item3': 5,
        },

    'user2':{
        'item1': 4,
        'item2': 5,
        'item3': 5,
        }
}

assert 0.999 < sim_pearson(critics['user1'], critics['user1']) < 1.0001

print('Your example:', sim_pearson(critics['user1'], critics['user2']))
print('Another example:', sim_pearson({1: 1, 2: 2, 3: 3}, {1: 4, 2: 0, 3: 1}))

1.0あなたの例では、ベクトル（-4 / 3、2 / 3、2 / 3）と（-2 / 3、1 / 3、1 / 3）が平行であるため、ピアソン係数が同じであることに注意してください。

于 2009-09-14T20:12:23.023 に答える