search

RUCAIBox / RecSysDatasets

This is a repository of public data sources for Recommender Systems (RS).
https://recbole.io/
844 stars 132 forks source link

推荐任务问题 #108

Open luanxian1 opened 2 years ago

luanxian1 commented 2 years ago

我想请教一下这个代码能否将BPR任务改成评分预测任务

Sherry-XLL commented 1 year ago

@luanxian1 您好，RecSysDatasets 是一个数据集的 repo，不涉及具体的排序或评分预测任务。

在 RecBole 中，若您需要将 BPR 的排序任务改为评分预测任务，则需要按照 Label of data 将数据的标签设置为评分 (如 rating)，并在模型中修改数据输入的 InputType 以及计算损失函数的函数 calculate_loss。

HowardZJU commented 1 year ago

@luanxian1 您好，RecSysDatasets 是一个数据集的 repo，不涉及具体的排序或评分预测任务。

在 RecBole 中，若您需要将 BPR 的排序任务改为评分预测任务，则需要按照 Label of data 将数据的标签设置为评分 (如 rating)，并在模型中修改数据输入的 InputType 以及计算损失函数的函数 calculate_loss。

你好，感谢解答！我最近也有评分预测任务的需求，请问如果按照这种方式处理后，trainer中的label是truncate后的，还是没有truncate的呢？能不能直接给模型输入不truncate的评分，构造一个回归任务？