RUCAIBox/RecBole推荐系统框架快速入门指南-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00891/article/details/148552750

RUCAIBox/RecBole推荐系统框架快速入门指南

RecBole RecBole - 这是一个关于推荐系统的开源项目，包含了一些关于推荐算法、协同过滤、Java 语言的示例和教程。适用于推荐系统、协同过滤、Java 语言编程等场景。项目地址: https://gitcode.com/gh_mirrors/re/RecBole

前言

RUCAIBox/RecBole是一个功能强大的推荐系统框架，为研究人员和开发者提供了构建、训练和评估推荐模型的完整工具链。本文将详细介绍如何使用该框架进行通用推荐任务的快速开发，以BPR模型在ml-100k数据集上的应用为例。

准备工作

数据集准备

RecBole内置了多个常用数据集，包括ml-100k（MovieLens 100K数据集），可以直接使用。该数据集包含：

943位用户
1682部电影
约10万条评分记录

对于自定义数据集，需要按照特定格式准备数据，包括用户ID、物品ID和交互信息等基本字段。

配置文件详解

RecBole使用YAML格式的配置文件来统一管理各种参数设置。下面我们分解配置文件的各个部分：

1. 数据集配置

USER_ID_FIELD: user_id
ITEM_ID_FIELD: item_id
load_col:
    inter: [user_id, item_id]

USER_ID_FIELD和ITEM_ID_FIELD指定用户和物品的ID字段名
load_col定义需要加载的数据列

2. 模型配置

以BPR模型为例：

embedding_size: 64

BPR(Bayesian Personalized Ranking)是一种经典的推荐算法，它通过学习用户和物品的潜在特征向量，并优化这些向量的排序来产生推荐。

3. 训练与评估配置

epochs: 500
train_batch_size: 4096
eval_batch_size: 4096
train_neg_sample_args:
    distribution: uniform
    sample_num: 1
eval_args:
    group_by: user
    order: RO
    split: {'RS': [0.8,0.1,0.1]}
    mode: full
metrics: ['Recall', 'MRR', 'NDCG', 'Hit', 'Precision']
topk: 10
valid_metric: MRR@10

关键参数说明：

epochs: 训练轮数
batch_size: 批处理大小
neg_sample_args: 负采样策略
eval_args: 评估策略
metrics: 评估指标
topk: 推荐列表长度

两种运行方式

1. 通过API运行

创建Python脚本(如run.py):

from recbole.quick_start import run_recbole
run_recbole(model='BPR', dataset='ml-100k', config_file_list=['test.yaml'])

执行命令:

python run.py

2. 通过命令行运行

直接使用框架提供的运行脚本:

python run_recbole.py --model=BPR --dataset=ml-100k --config_files=test.yaml

如需修改参数，如embedding_size:

python run_recbole.py --model=BPR --dataset=ml-100k --config_files=test.yaml --embedding_size=100

结果解读

运行完成后，控制台会输出类似以下信息：

valid result:
recall@10 : 0.2162
mrr@10 : 0.3752
ndcg@10 : 0.2284
hit@10 : 0.7508
precision@10 : 0.1602

test result:
recall@10 : 0.2523
mrr@10 : 0.4855
ndcg@10 : 0.292
hit@10 : 0.7953
precision@10 : 0.1962

指标说明：