【发布时间】:2020-05-04 05:27:42
【问题描述】:
总的来说,我对 DataScience/Pandas 很陌生。我主要关注this guide 的推荐系统。
我的主要区别是我的电影数据从一个表开始,我想从该列表中找到与特定电影最相似的电影。
然后我将其转换为稀疏数据帧,其中可以显示用户对每部电影的评分:
rating_with_totalRatingCount = rating_with_totalRatingCount.drop_duplicates(['userID', 'movieTitle'])
ratingPivot = rating_with_totalRatingCount.pivot(index = 'movieTitle', columns = 'userID', values = 'userRatingScore').fillna(0).T
最后我应用点积规则来获取预测数据框:
U, sigma, Vt = svds(ratingPivot, k = 50)
sigma = np.diag(sigma)
all_user_predicted_ratings = np.dot(np.dot(U, sigma), Vt) + user_ratings_mean.values.reshape(-1, 1)
preds_df = pd.DataFrame(all_user_predicted_ratings, columns = ratingPivot.columns)
我的问题是,我如何使用这个预测数据框来为某人提供与 Python 中的特定电影最相似的电影?
【问题讨论】:
标签: python numpy matrix prediction recommender-systems