RecBole
RecBole copied to clipboard
请问S3Rec如何使用多种特征?
首先感谢贵团队开发了这个方便好用的工具包,我在尝试使用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 您好,感谢您的关注!
目前RecBole框架下item_attribute
字段只能指定单个特征进行预训练。如果你想充分复现原论文的结果,可以参考官网给出的源码S3Rec。