機器學習第一篇(下)
阿新 • • 發佈:2018-11-11
實戰演練 ——MovieLens
第一步:獲取資料集
第二步:解壓檔案後,將u.item以及u.data新增到專案中,加入方法loadMovieLens
def loadMovieLens(path='data/movielens'): movies={} for line in open(path+'/u.item',encoding='UTF-8'): (id,title)=line.split('|')[0:2] movies[id]=title prefs={ } for line in open(path+'/u.data'): (user,movieid,rating,ts)=line.split('\t') prefs.setdefault(user,{ }) prefs[user][movies[movieid]]=float(rating) return prefs
這裡要注意open(path+'/u.item',encoding='UTF-8'),若不加encoding='UTF-8',會報錯,其他的就很簡單啦
不足:在這一篇中有一個基於物品推薦的還不是很明白,過段時間爭取搞懂