RecBole
RecBole copied to clipboard
【序列推荐】测试数据集的session id不在训练数据集中,请问这种情况RecBole能处理吗?
具体情况是这样,数据集分两种: train数据集,session id是从0-199999, test数据集,session id是从200000-250000
现在我用train数据集训练得到序列模型SRGNN。
然后我想用得到的模型对test数据集做预测,(注意模型并没有见过test数据集),应该怎么做?
@ralgond 感谢您的关注!
train和test数据集中session id不重合,可以在同一阶段输入train和test数据集。将训练和测试分开会使模型ID域无法对齐。使用自定义划分的序列数据集可以通过设置benchmark_filename
来实现,更多的细节可以参考#issue 1069。