RUCAIBox / RecBole

A unified, comprehensive and efficient recommendation library
https://recbole.io/
MIT License
3.27k stars 590 forks source link

[💡SUG] 我想请教怎么添加物品的类别属性信息到模型中? #2059

Open kmr666 opened 3 weeks ago

kmr666 commented 3 weeks ago

您好,我现在想跑一个对比算法,他的算法用到了物品的类别信息,并且用到了dgl库来构造图,现在我想用recbole提供的三个数据集分别是,yelp,gowalla,amazon_movies来跑这个算法,我有以下两个问题: 1)我使用的是recbole处理后的数据集,这里的gowalla只有inter数据,其中没有物品的属性信息,请问如何添加gowalla数据集物品的类别信息?还是说gowalla数据集没有提供item的类别信息? 2)如果我想根据目前的处理后的.user,.item,.inter文件来跑对比算法,对用户和物品的remap操作以及对相应的物品类别属性的remap是否有脚本处理呢?因为我想把user和item交互以及每个物品的类别信息也提取出来作为辅助信息,模型也需要根据这个属性信息进行计算,请问是否有相应的脚本实现?