推荐系统物质扩散代码python

推荐系统当中的物质扩散算法，使用python语言编写，有效快捷

# coding:utf-8\

from operator import itemgetter

‘‘‘
dataSet格式{user:[item...]...}
dataSetT格式{item:[user...]...}
‘‘‘
def Prob（dataSet dataSetT User n）:
    ‘‘‘Step1------物品拥有的资源值扩散到用户‘‘‘
    UserResource = {}
    for item in dataSet[User]:
        kItem = len（dataSetT[item]）
        for user in dataSetT[item]:
                UserResource[user] = 1.0 / float（kItem） + UserResource.setdefault（user 0）

    ‘‘‘Step2----用户得到的资源值扩散到物品‘‘‘
    ItemResource = {}
    for user in UserResource.keys（）:
        energy = UserResource[user]
        kUser = len（dataSet[user]）
        for item in dataSet[user]:
            ItemResource[item] = energy / float（kUser） + ItemResource.setdefault（item 0）

    for item in dataSet[User]:
        ItemResource[item] = 0.0

    # 返回能量最多的前n个物品推荐给目标用户
    return sorted（ItemResource.items（） key=itemgetter（1） reverse=True）[0:n]

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件        992  2018-06-26 18:51  物质扩散.py

----------- ---------  ---------- -----  ----

                  992                    1