RecBole icon indicating copy to clipboard operation
RecBole copied to clipboard

【序列推荐】测试数据集的session id不在训练数据集中,请问这种情况RecBole能处理吗?

Open ralgond opened this issue 1 year ago • 1 comments

具体情况是这样,数据集分两种: train数据集,session id是从0-199999, test数据集,session id是从200000-250000

现在我用train数据集训练得到序列模型SRGNN。

然后我想用得到的模型对test数据集做预测,(注意模型并没有见过test数据集),应该怎么做?

ralgond avatar Dec 05 '23 14:12 ralgond

@ralgond 感谢您的关注! train和test数据集中session id不重合,可以在同一阶段输入train和test数据集。将训练和测试分开会使模型ID域无法对齐。使用自定义划分的序列数据集可以通过设置benchmark_filename来实现,更多的细节可以参考#issue 1069

Fotiligner avatar Dec 20 '23 03:12 Fotiligner