协同过滤算法实现

来源:互联网 发布:绝爱后宫我知帝王心txt 编辑:程序博客网 时间:2024/05/01 23:37

用户物品推荐

import pandas as pd 

import numpy as np
a = pd.DataFrame({'u1':{'l1':10,'l3':3},'u2':{'l2':3},'u3':{'l1':5,'l2':4}})
b=a.fillna(0)

c= b.T

#求出用户相似度

from sklearn.metrics.pairwise import pairwise_distances

u = pairwise_distances(c,metric='cosine')




#实现上式子

m = c.mean(axis=1)[:,np.newaxis]*np.ones(c.shape)
n =np.array([np.abs(u).sum(axis=1)]).T
r =c-m
p = m+ u.dot(r)/n