Open zw81929 opened 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支持真的太棒了
某些情况下,user_id并不需要关注,我需要实现的使 [item_id,item_id,item_id,item_id,....] 能预测出新的 [item_id,item_id,item_id,item_id,....]
目前的 study 的demo 都是最后计算的user_id