search

Coder-Yu / QRec

QRec: A Python Framework for quick implementation of recommender systems (TensorFlow Based)
1.59k stars 406 forks source link

Douban-Book数据集 #242

Closed malajikuai closed 2 years ago

malajikuai commented 2 years ago

余老师您好： SimGCL没有提供Douban-book的数据集，请问是否能提供下？

Coder-Yu commented 2 years ago

你好，参见 https://github.com/Coder-Yu/SELFRec/tree/main/dataset/douban-book

malajikuai commented 2 years ago

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

Coder-Yu commented 2 years ago

ssl_reg 0.1 droprate 0.1 tau 0.2

malajikuai commented 2 years ago

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

Coder-Yu commented 2 years ago

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

malajikuai commented 2 years ago

谢谢老师。我的参数设置如下：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 commented 2 years ago

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

malajikuai commented 2 years ago

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

Coder-Yu commented 2 years ago

^_^ 不用谢