LehengTHU / AdvInfoNCE

[NeurIPS 2023] The implementation of paper "Empowering Collaborative Filtering Generalization via Principled Adversarial Contrastive Loss"
MIT License
18 stars 2 forks source link

关于kuaiRec2的一个问题 #2

Open user683 opened 6 months ago

user683 commented 6 months ago

作者您好,在阅读您代码时,发现你对快手这个文件也分别生成的test_id和test_ood, 请问这两个数据集分别是怎么生成的呢?test_id是不是从处理后的kuairec2里的big_matrix里在切分一部分出来呢?而test_ood就是原来的快里的small_matrix呢?

LehengTHU commented 6 months ago

感谢您对工作的关注。test_id和test_ood实际上是同一个文件,就是small_matrix。抱歉对您造成了困扰。这样处理的原因是为了和其他数据集的评估方式不产生冲突,避免写额外的评估代码(其他数据集都有3个id和1个ood)