案例--基于协同过滤的电影推荐

最新推荐文章于 2024-03-16 16:58:28 发布

IT瘾君

最新推荐文章于 2024-03-16 16:58:28 发布

阅读量1.4k

点赞数

分类专栏：推荐系统-大数据文章标签： python 机器学习深度学习

本文链接：https://blog.csdn.net/u012441595/article/details/122260325

版权

推荐系统-大数据专栏收录该内容

23 篇文章 0 订阅

订阅专栏

python编程快速上手（持续更新中…）

学习目标

应用基于用户的协同过滤实现电影评分预测
应用基于物品的协同过滤实现电影评分预测

User-Based CF 预测电影评分

数据集下载

下载地址：MovieLens Latest Datasets Small
提取码：c5fe
数据量小，便于我们单机使用和运行

加载ratings.csv，转换为用户-电影评分矩阵并计算用户之间相似度

1.加载数据

# 导入包
import pandas as pd
import numpy as np

# 定义类型
dtype = {"userId": np.int32, "movieId": np.int32, "rating": np.float32}

# 加载数据
ratings = pd.read_csv('../data/ml-latest-small/ratings.csv')
ratings

在这里插入图片描述

2.构建透视表，用户和电影之间关系

# 透视表，将电影ID转换为列名称，转换成为一个User-Movie的评分矩阵
ratings_matrix = ratings.pivot_table(index=["userId"], columns=["movieId"],values="rating")
ratings_matrix

在这里插入图片描述

3.计算用户之间相似度

# 计算用户之间相似度
user_similar = ratings_matrix.T.corr()
user_similar

在这里插入图片描述

4.预测用户对物品的评分（以用户1对电影1评分为例）

在这里插入图片描述

# 预测用户1和物品1评分
similar_users = user_similar[1].drop([1]).dropna()
similar_users

在这里插入图片描述

5.相似用户筛选规则：正相关的用户

# 相似用户筛选规则：正相关的用户
similar_users = similar_users.where(similar_users>0).dropna()
similar_users

在这里插入图片描述

6. 从用户1的近邻相似用户中筛选出对物品1有评分记录的近邻用户

ids = set(ratings_matrix[1].dropna().index)&set(similar_users.index)
ids

在这里插入图片描述

# 
finally_similar_users = similar_users.loc[list(ids)]
finally_similar_users

在这里插入图片描述

7.计算用户1对商品1评分预测

# 计算用户1对商品1评分预测
numerator = 0    # 评分预测公式的分子部分的值
denominator = 0    # 评分预测公式的分母部分的值
for sim_uid, similarity in finally_similar_users.iteritems():
    # 近邻用户的评分数据
    sim_user_rated_movies = ratings_matrix.loc[sim_uid].dropna()
    # 近邻用户对iid物品的评分
    sim_user_rating_for_item = sim_user_rated_movies[1]
    # 计算分子的值
    numerator += similarity * sim_user_rating_for_item
    # 计算分母的值
    denominator += similarity
# 4 计算预测的评分值
predict_rating = numerator/denominator
print("预测出用户<%d>对电影<%d>的评分：%0.2f" % (1, 1, predict_rating))