search

RUCAIBox / RecBole-CDR

This is a library built upon RecBole for cross-domain recommendation algorithms
MIT License
82 stars 12 forks source link

[Evaluation]能否同时对源域和目标域进行？ #65

Open Little-wen opened 6 months ago

Little-wen commented 6 months ago

我想在此框架上新增一个自己的双目标跨域推荐模型，但是框架提供的evaluation似乎只针对目标域。能否实现同时对源域和目标域进行evaluate？

Wicknight commented 5 months ago

@Little-wen 您好！感谢您对RecBole-CDR的关注！RecBole-CDR的评测框架继承自RecBole，目前仅支持对目标域做评测，如果您想要同时对源域和目标域做评测，您需要重写Trainer中的evaluate函数，同时保存两个数据域的data_struct，分别进行评测。