RecSysDatasets: 推荐系统数据集大全

RecSysDatasets

RecSysDatasets: 推荐系统研究的数据宝库

在推荐系统领域,高质量的数据集对于算法的开发和评估至关重要。RecSysDatasets项目为研究人员提供了一个全面的推荐系统公开数据源库,涵盖了电商、广告、电影、音乐等多个领域的数据集。本文将详细介绍该项目的主要内容和特点,为推荐系统研究者提供数据资源指南。

RecSysDatasets是由中国人民大学AI Box团队开源的一个推荐系统数据集仓库。该项目的目标是收集和整理各类公开可用的推荐系统数据集,并提供统一的数据格式和使用方法,方便研究人员进行算法开发和性能评估。

项目的主要特点包括:

通过使用RecSysDatasets,研究人员可以方便地获取各类推荐系统数据集,并利用RecBole库进行快速的算法实验。这大大降低了数据获取和处理的成本,有助于推动推荐系统研究的发展。

RecSysDatasets收录的数据集涵盖了推荐系统的多个应用场景,主要可以分为以下几类:

电商是推荐系统最重要的应用领域之一。RecSysDatasets收录了多个知名的电商数据集:

Amazon系列数据集
- Amazon Review Data: 包含亚马逊产品评论和元数据,有2014、2018和2023三个版本
- Amazon_M2: 来自6个不同语言地区的匿名用户会话数据
Alibaba-iFashion: 阿里巴巴提供的时尚穿搭数据集
Epinions: 从Epinions.com网站收集的用户评论数据
Yelp: 包含商家、评论和用户数据,有多个版本
Tmall: 天猫提供的用于IJCAI16比赛的数据集
DIGINETICA: 从电商搜索引擎日志中提取的用户会话数据
YOOCHOOSE: 用于RecSys Challenge 2015的数据集
Retailrocket: 来自真实电商网站的原始数据
Ta Feng: 中国台湾杂货店的交易数据

这些数据集涵盖了用户评分、浏览历史、购买行为等多种交互类型,为研究不同场景下的推荐算法提供了丰富的数据资源。

Amazon数据集示意图

在线广告是推荐系统的另一个重要应用。RecSysDatasets收录了以下广告相关数据集:

这些数据集主要用于研究广告点击率(CTR)预测问题,包含丰富的用户和广告特征信息。

基于位置的社交网络(LBSN)是推荐系统的一个新兴应用领域。RecSysDatasets收录了两个经典的签到数据集:

这些数据集可用于研究基于位置的推荐算法。

电影推荐是推荐系统研究的经典问题。RecSysDatasets收录了多个电影相关数据集:

这些数据集包含用户对电影的评分、评论等信息,可用于研究协同过滤等经典推荐算法。

MovieLens数据集示意图

音乐推荐是另一个重要的应用场景。RecSysDatasets收录了以下音乐相关数据集:

这些数据集涵盖了用户的音乐收听行为、社交关系、音乐元数据等多方面信息,为音乐推荐研究提供了丰富的数据资源。

RecSysDatasets还收录了两个经典的图书推荐数据集:

这些数据集可用于研究图书推荐系统。

除了上述主要类别,RecSysDatasets还收录了游戏、动漫、图片、笑话、运动、新闻、食品、饮料、服装等多个领域的特色数据集,为各种垂直领域的推荐系统研究提供了数据支持。

为了方便研究人员使用这些数据集,RecSysDatasets项目提供了详细的使用说明:

数据格式转换 RecSysDatasets支持将原始数据集转换为RecBole库定义的统一格式(atomic files)。用户可以通过以下两种方式获取转换后的数据:
- 下载原始数据集,使用项目提供的转换工具进行处理
- 直接下载项目预处理好的atomic files格式数据
数据集信息统计
项目提供了各数据集的详细统计信息,包括用户数、物品数、交互数、稀疏度等,方便研究人员选择合适的数据集。
与RecBole库集成转换为atomic files格式后,用户可以直接使用RecBole库加载数据集,快速进行各种推荐算法的实验。
持续更新项目会持续关注和收录新发布的推荐系统数据集,用户可以通过GitHub随时获取最新的数据资源。