机器学习推荐算法案例代码
时间: 2023-06-23 19:55:48 浏览: 62
以下是一个基于Python的协同过滤推荐算法案例代码:
```python
import numpy as np
import pandas as pd
# 加载数据
ratings_data = pd.read_csv('ratings.csv')
movie_names = pd.read_csv('movies.csv')
# 合并数据
movie_data = pd.merge(ratings_data, movie_names, on='movieId')
# 创建评分矩阵
ratings_matrix = movie_data.pivot_table(index='userId', columns='title', values='rating')
# 计算相似度矩阵
movie_similarity = ratings_matrix.corr(method='pearson', min_periods=10)
# 推荐函数
def get_similar_movies(movie_name, user_rating):
similar_score = movie_similarity[movie_name]*(user_rating-2.5)
similar_score = similar_score.sort_values(ascending=False)
return similar_score
# 测试推荐函数
action_lover = [('2 Fast 2 Furious (Fast and the Furious 2, The) (2003)',5), ('12 Years a Slave (2013)',3), ('300 (2007)',4), ('28 Days Later (2002)',4)]
similar_movies = pd.DataFrame()
for movie, rating in action_lover:
similar_movies = similar_movies.append(get_similar_movies(movie, rating), ignore_index=True)
print(similar_movies.sum().sort_values(ascending=False))
```
该代码使用了协同过滤推荐算法,基于用户的评分数据和电影相似度矩阵,为用户推荐与他们喜欢的电影相似的其他电影。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)