快捷導(dǎo)航

簡單的python協(xié)同過濾程序?qū)嵗a

更新時間：2018年01月31日 11:11:20 作者：葉落花開

這篇文章主要介紹了簡單的python協(xié)同過濾程序，分享了相關(guān)代碼示例，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下

本文研究的主要是python協(xié)同過濾程序的相關(guān)內(nèi)容，具體介紹如下。

關(guān)于協(xié)同過濾的一個最經(jīng)典的例子就是看電影，有時候不知道哪一部電影是我們喜歡的或者評分比較高的，那么通常的做法就是問問周圍的朋友，看看最近有什么好的電影推薦。在問的時候，都習(xí)慣于問跟自己口味差不多的朋友，這就是協(xié)同過濾的核心思想。

這個程序完全是為了應(yīng)付大數(shù)據(jù)分析與計算的課程作業(yè)所寫的一個小程序，先上程序，一共55行。不在意細節(jié)的話，55行的程序已經(jīng)表現(xiàn)出了協(xié)同過濾的特性了。就是對每一個用戶找4個最接近的用戶，然后進行推薦，在選擇推薦的時候是直接做的在4個用戶中選擇該用戶item沒包括的，當然這里沒限制推薦數(shù)量，個人覺得如果要提高推薦準確率的畫，起碼，1，要對流行的item進行處理。2，將相鄰的四個用戶的item進行排序，從多到少的進行推薦。程序所用的數(shù)據(jù)是movielens上的（http://grouplens.org/datasets/movielens）。相似度的計算也很簡單，直接用了交集和差集的比值。好吧，上程序

#coding utf-8
import os
import sys
import re

f1=open("/home/alber/data_base/bigdata/movielens_train_result.txt",'r')  #讀取train文件，已經(jīng)處理成每一行代表一位用戶的item，項之間用空格。
f2=open("/home/alber/data_base/bigdata/movielens_train_result3.txt",'a')
txt=f1.readlines()
contxt=[]
f1.close()
userdic={}
for line in txt:
  line_clean=" ".join(line.split())
  position=line_clean.index(",")
  ID=line_clean[0:position]
  item=line_clean[position+1:]
  userdic.setdefault(ID,item)
  if len(item)>=5:           #對觀影量少于5的用戶不計入相似性計算的范圍
    contxt.append(item)
for key in userdic.keys():        #計算每位用戶的4個最相似用戶
  ID_num=key
  value=userdic[key]
   user_item=value.split(' ')
   Sim_user=[]
   for lines in contxt:
     lines_clean=lines.split(' ')
     intersection=list(set(lines_clean).intersection(set(user_item)))
     lenth_intersection=len(intersection)
     difference=list(set(lines_clean).difference(set(user_item)))
     lenth_difference=len(difference)
     if lenth_difference!=0:                     
       Similarity=float(lenth_intersection)/lenth_difference          #交集除以差集作為相似性的判斷條件
       Sim_user.append(Similarity)
     else:
       Sim_user.append("0")
   Sim_user_copy=Sim_user[:]
   Sim_user_copy.sort()
   Sim_best=Sim_user_copy[-4:]
   position1=Sim_user.index(Sim_best[3])
   position2=Sim_user.index(Sim_best[2])
   position3=Sim_user.index(Sim_best[1])
   position4=Sim_user.index(Sim_best[0])
   if position1!=0 and position2!=0 and position3!=0 and position4!=0:
     recommender=userdic[str(position1)]+" "+userdic[str(position2)]+" "+userdic[str(position3)]+" "+userdic[str(position4)] #將4位用戶的看過的電影作為推薦 
  else:
    recommender="none"   
  reco_list=recommender.split(' ')
  recomm=[]
  for good in reco_list:
    if good not in user_item:
      recomm.append(good)
    else:
      pass
  f2.write((" ".join(recomm)+"\n"))
f2.close()

總結(jié)

以上就是本文關(guān)于簡單的python協(xié)同過濾程序?qū)嵗a的全部內(nèi)容，希望對大家有所幫助。感興趣的朋友可以繼續(xù)參閱本站其他相關(guān)專題，如有不足之處，歡迎留言指出。感謝朋友們對本站的支持！

您可能感興趣的文章: