首页 文章详情

surprise,一个有趣的 Python 库!

日常学python | 154 2024-06-18 13:11 0 0 0
UniSMS (合一短信)
0904cf5ae8f4b8dd9fd4ae9f3cd8c579.webp
更多Python学习内容: ipengtao.com

大家好,今天为大家分享一个有趣的 Python 库 - surprise。

Github地址:https://github.com/NicolasHug/Surprise


在推荐系统领域,协同过滤是最常用的方法之一。Python的Surprise库是一个专门用于构建和分析推荐系统的开源库。它提供了多种协同过滤算法,方便开发者快速构建、评估和优化推荐系统。Surprise库不仅支持常见的算法,还提供了易用的API和丰富的功能,帮助开发者高效地处理推荐任务。本文将详细介绍Surprise库,包括其安装方法、主要特性、基本和高级功能,以及实际应用场景,帮助全面了解并掌握该库的使用。

安装

要使用Surprise库,首先需要安装它。可以通过pip工具方便地进行安装。

以下是安装步骤:

      
      pip install scikit-surprise

安装完成后,可以通过导入surprise库来验证是否安装成功:

      
      import surprise
print("Surprise库安装成功!")

特性

  1. 多种推荐算法:支持用户-物品协同过滤、矩阵分解、基于邻域的方法等多种推荐算法。
  2. 易于评估:内置多种评估指标和交叉验证方法,方便评估模型性能。
  3. 灵活的数据导入:支持从多种数据格式导入数据,包括内存中的数据、文件和Pandas DataFrame。
  4. 自动调参:支持超参数优化和网格搜索,帮助找到最佳模型参数。
  5. 易用的API:提供简洁、易用的API,快速上手推荐系统开发。

基本功能

数据加载

Surprise库支持从文件和内存中加载数据。

以下是一个简单的示例,演示如何从文件中加载数据:

      
      from surprise import Dataset

# 加载内置的movielens数据集
data = Dataset.load_builtin('ml-100k')

# 获取训练集
trainset = data.build_full_trainset()
print("数据加载成功!")

构建推荐模型

Surprise库提供了多种推荐算法,以下是使用SVD算法构建推荐模型的示例:

      
      from surprise import SVD
from surprise import Dataset

# 加载数据
data = Dataset.load_builtin('ml-100k')
trainset = data.build_full_trainset()

# 使用SVD算法
algo = SVD()
algo.fit(trainset)
print("模型训练成功!")

预测评分

训练完成后,可以使用模型进行评分预测。

以下是一个示例,演示如何进行评分预测:

      
      # 预测用户对某个物品的评分
uid = str(196)  # 用户ID
iid = str(302)  # 物品ID
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

模型评估

Surprise库提供了多种评估指标和交叉验证方法,以下是使用RMSE评估模型性能的示例:

      
      from surprise import accuracy
from surprise.model_selection import train_test_split

# 划分训练集和测试集
trainset, testset = train_test_split(data, test_size=0.25)

# 训练模型
algo.fit(trainset)

# 预测
predictions = algo.test(testset)

# 计算RMSE
rmse = accuracy.rmse(predictions)
print(f"模型RMSE: {rmse}")

高级功能

自定义数据加载

Surprise库支持从Pandas DataFrame中加载数据。

以下是一个示例,演示如何从DataFrame加载数据:

      
      import pandas as pd
from surprise import Dataset
from surprise import Reader

# 创建DataFrame
data = {
    'userID': [111223],
    'itemID': [123123],
    'rating': [534453]
}
df = pd.DataFrame(data)

# 定义数据格式
reader = Reader(rating_scale=(15))

# 加载数据
data = Dataset.load_from_df(df[['userID''itemID''rating']], reader)
trainset = data.build_full_trainset()
print("数据加载成功!")

超参数优化

Surprise库支持超参数优化,以下是使用网格搜索进行超参数优化的示例:

      
      from surprise.model_selection import GridSearchCV

# 定义参数网格
param_grid = {'n_factors': [50100], 'n_epochs': [2030], 'lr_all': [0.0020.005]}
gs = GridSearchCV(SVD, param_grid, measures=['rmse'], cv=3)

# 执行网格搜索
gs.fit(data)

# 输出最佳参数
print("最佳参数:", gs.best_params['rmse'])

使用不同的推荐算法

Surprise库支持多种推荐算法,以下是使用KNNBasic算法的示例:

      
      from surprise import KNNBasic

# 使用KNNBasic算法
algo = KNNBasic()
algo.fit(trainset)

# 预测评分
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

交叉验证

Surprise库提供了便捷的交叉验证功能,以下是一个使用交叉验证评估模型性能的示例:

      
      from surprise.model_selection import cross_validate

# 执行交叉验证
results = cross_validate(algo, data, measures=['RMSE''MAE'], cv=5, verbose=True)
print("交叉验证结果:", results)

实际应用场景

电影推荐系统

假设在开发一个电影推荐系统,希望根据用户的历史评分推荐电影,可以使用Surprise库实现这一功能。

      
      from surprise import SVD
from surprise import Dataset
from surprise.model_selection import train_test_split

# 加载数据
data = Dataset.load_builtin('ml-100k')

# 划分训练集和测试集
trainset, testset = train_test_split(data, test_size=0.25)

# 使用SVD算法
algo = SVD()
algo.fit(trainset)

# 预测评分
uid = str(196)  # 用户ID
iid = str(302)  # 物品ID
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

电商产品推荐

假设在开发一个电商平台,希望根据用户的购买历史推荐商品,可以使用Surprise库实现这一功能。

      
      import pandas as pd
from surprise import Dataset
from surprise import Reader
from surprise import SVD
from surprise.model_selection import train_test_split

# 创建DataFrame
data = {
    'userID': [111223],
    'itemID': [123123],
    'rating': [534453]
}
df = pd.DataFrame(data)

# 定义数据格式
reader = Reader(rating_scale=(15))

# 加载数据
data = Dataset.load_from_df(df[['userID''itemID''rating']], reader)

# 划分训练集和测试集
trainset, testset = train_test_split(data, test_size=0.25)

# 使用SVD算法
algo = SVD()
algo.fit(trainset)

# 预测评分
uid = str(1)  # 用户ID
iid = str(2)  # 物品ID
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

音乐推荐系统

假设在开发一个音乐推荐系统,希望根据用户的听歌历史推荐歌曲,可以使用Surprise库实现这一功能。

      
      import pandas as pd
from surprise import Dataset
from surprise import Reader
from surprise import KNNBasic
from surprise.model_selection import train_test_split

# 创建DataFrame
data = {
    'userID': [111223],
    'itemID': [123123],
    'rating': [534453]
}
df = pd.DataFrame(data)

# 定义数据格式
reader = Reader(rating_scale=(15))

# 加载数据
data = Dataset.load_from_df(df[['userID','itemID''rating']], reader)

# 划分训练集和测试集
trainset, testset = train_test_split(data, test_size=0.25)

# 使用KNNBasic算法
algo = KNNBasic()
algo.fit(trainset)

# 预测评分
uid = str(1)  # 用户ID
iid = str(2)  # 物品ID
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

个性化内容推荐

假设在开发一个内容平台,希望根据用户的历史偏好推荐个性化内容,可以使用Surprise库实现这一功能。

      
      import pandas as pd
from surprise import Dataset
from surprise import Reader
from surprise import SVD
from surprise.model_selection import train_test_split

# 创建DataFrame
data = {
    'userID': [111223],
    'itemID': [123123],
    'rating': [534453]
}
df = pd.DataFrame(data)

# 定义数据格式
reader = Reader(rating_scale=(15))

# 加载数据
data = Dataset.load_from_df(df[['userID''itemID''rating']], reader)

# 划分训练集和测试集
trainset, testset = train_test_split(data, test_size=0.25)

# 使用SVD算法
algo = SVD()
algo.fit(trainset)

# 预测评分
uid = str(1)  # 用户ID
iid = str(2)  # 物品ID
pred = algo.predict(uid, iid)
print(f"预测评分: {pred.est}")

总结

Surprise库是一个功能强大且易于使用的推荐系统开发工具,能够帮助开发者高效地构建、评估和优化推荐系统。通过支持多种推荐算法、灵活的数据加载方式、易于评估和超参数优化等特性,Surprise库能够满足各种推荐系统开发需求。本文详细介绍了Surprise库的安装方法、主要特性、基本和高级功能,以及实际应用场景。希望本文能帮助大家全面掌握Surprise库的使用,并在实际项目中发挥其优势。

如果你觉 得文章还不错 ,请大家 点赞、分享、留言 下,因为这将是我持续输出更多 优质文章的最强动力!

更多Python学习内容: ipengtao.com


果想要系统学习Python、Python问题咨询,或者考虑做一些工作以外的副业,都可以扫描二维码添加微信,围观朋友圈一起交流学习。

1952db828d6120aceb922f7b74930947.webp

我们还为大家准备了Python资料和副业项目合集,感兴趣的小伙伴快来找我领取一起交流学习哦!

11d769dcec2899315e9780afad2c9461.webp

往期推荐



历时一个月整理的 Python 爬虫学习手册全集PDF(免费开放下载)

全网最全 Pandas的入门与高级教程全集,都在这里了!(PDF下载)

120道Python面试题.pdf ,完全版开放下载

Beautiful Soup快速上手指南,从入门到精通(PDF下载)

120道Python面试题.pdf ,完全版开放下载

Python办公自动化完全指南(免费PDF)

7951f72f5fb8fb949179da00cd1938ac.webp“阅读原文”一起来充电吧!
good-icon 0
favorite-icon 0
收藏
回复数量: 0
    暂无评论~~
    Ctrl+Enter