RUCAIBox / RecBole

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

[💡SUG] 目前的bert4rec中能实现seq -> seq 么? #2056

Open zw81929 opened 1 month ago

zw81929 commented 1 month ago

某些情况下,user_id并不需要关注,我需要实现的使 [item_id,item_id,item_id,item_id,....] 能预测出新的 [item_id,item_id,item_id,item_id,....]

目前的 study 的demo 都是最后计算的user_id

zw81929 commented 1 month ago

补充一下:现在在case_study 使用的torch.full_like 依然是user_id找item_id, 能不能单独有个接口,直接传入一个tensor[item_id,item_id,item_id] ,这个list可能是随机的几个item_id组成,根据前面的item计算最后返回 K 个top item_id 类似这个项目 https://github.com/vatsalsaglani/bert4rec

最后,说实话,你们的yaml方式配置,多gpu支持真的太棒了