lulu0913 / VRKG4Rec

This is the PyTorch Implementation for our model VRKG4Rec (WSDM'23)
19 stars 1 forks source link

在论文对比实验中KGIN的参数 #3

Closed newfsyyy closed 1 year ago

newfsyyy commented 1 year ago

你好,请问KGIN的实验是如何进行的,我采用了MCCLK论文中固定采样方式8:2分别生成训练集和测试集,我的随机数种子为2023,然后进行训练,其中epoch设置为1000,batch为1024,dim为64,lr为1e-4,test_batch 128这些参数我没有改动,在多轮训练中RKG4Rec在Recall@20中能到达接近30%多的性能,但KGIN我尝试多个参数,在100次的epoch的Recall@20中总是维持在10%-12%之间,甚至会出现性能下降,我很想知道如何进行基线的实验,非常感谢