1. 程式人生 > >機器學習第一篇(下)

機器學習第一篇(下)

實戰演練 ——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',會報錯,其他的就很簡單啦

 

不足:在這一篇中有一個基於物品推薦的還不是很明白,過段時間爭取搞懂