37,720
社区成员
发帖
与我相关
我的任务
分享
def loadMovieLens(path='E://python/movielens'):
# Get movie titles
movies = {}
for line in open(path + '/u.item'):
(id, title) = line.split('|')[0:2]
movies[id] = title
# Load data
prefs = {}
for line in open(path + '/u.data'):
(user, movieid, rating, ts) = line.split('\t')
prefs.setdefault(user, {})
prefs[user][movies[id]] = float(rating)
return prefs