RUCAIBox / RecBole-CDR

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

[🐛BUG] BiTGCF模型鲁棒性 #53

Closed WYH-han closed 1 year ago

WYH-han commented 1 year ago

描述这个 bug BiTGCF模型中包含Feature Transfer部分,其中在公式(15)-(16)计算user(item)-related weight factor时,需要考虑多跳后可能出现某些节点的邻居为0,即公式(15)中分母可能为0的情况,则会导致模型训练中出现“nan”。报错如下: image image

预期 为了避免“除0”问题,针对项目代码中的user_laplace和item_laplace,即分母,添加一个极小数以避免出现“除0”问题。

image

屏幕截图 image

Wicknight commented 1 year ago

@WYH-han 感谢您的提醒,我们已经在 #56 中修复了这个问题!