RecBole icon indicating copy to clipboard operation
RecBole copied to clipboard

请问S3Rec如何使用多种特征?

Open yfwang2021 opened this issue 2 years ago • 1 comments

首先感谢贵团队开发了这个方便好用的工具包,我在尝试使用S3Rec模型时遇到了一个问题:在S3Rec原论文中是可以同时使用多种特征进行Pretrain的,例如论文在Meituan数据上同时使用了category、location和keywords,但我发现Recbole实现的S3Rec似乎只能使用一种特征?文档中的相关参数和描述是“item_attribute (str) : The item features used as attributes for pre-training. Defaults to 'class' for ml-100k dataset.” 这里似乎只能传入一个特征,我尝试将参数设置为["feat1","feat2","feat3"]但会报错。请问我要如何使用多种特征呢?

yfwang2021 avatar May 03 '22 04:05 yfwang2021

@yfwang2021 您好,感谢您的关注! 目前RecBole框架下item_attribute字段只能指定单个特征进行预训练。如果你想充分复现原论文的结果,可以参考官网给出的源码S3Rec

Ethan-TZ avatar May 03 '22 16:05 Ethan-TZ