Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Douban-Book数据集 #242

Closed
malajikuai opened this issue Aug 25, 2022 · 9 comments
Closed

Douban-Book数据集 #242

malajikuai opened this issue Aug 25, 2022 · 9 comments

Comments

@malajikuai
Copy link

余老师您好:
SimGCL没有提供Douban-book的数据集,请问是否能提供下?

@Coder-Yu
Copy link
Owner

@malajikuai
Copy link
Author

好的,谢谢老师。请问老师能否帮忙提供下对照实验中SGL模型在该数据集上的参数?

@Coder-Yu
Copy link
Owner

ssl_reg 0.1 droprate 0.1 tau 0.2

@malajikuai
Copy link
Author

老师,这个数据集被拆分成train,test和trust,最后进行实验的时候训练集上用的train数据集,test的时候用的test数据集,trust数据集不使用对吧?

@Coder-Yu
Copy link
Owner

嗯, trust是给social recommendation的模型用的

@malajikuai
Copy link
Author

谢谢老师。我的参数设置如下:SGL=-n_layer 3 -lambda 0.1 -droprate 0.1 -augtype 1 -temp 0.2,然后结果的话:'Recall:0.1696415082289009\n','NDCG:0.15280217620274245\n'和论文中的实验结果:0.1732 0.1551 还是有一点差异的。

@Coder-Yu
Copy link
Owner

你试试合并train和test 然后用QRec自带的5折交叉验证应该能复现SGL的结果。目前这个划分是我后来随机划分的,可能会跟表上的结果稍有一点出入。我尝试了一下SGL跟你的结果一样,SimGCL是Recall:0.1741688500232818 NDCG:0.1543619201814785。

@malajikuai
Copy link
Author

谢谢老师,按照老师说的方法复现出来了。

@Coder-Yu
Copy link
Owner

^_^ 不用谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants