Applying NCL on Other GNN Backbones.

RUCAIBox / NCL

[WWW'22] Official PyTorch implementation for "Improving Graph Collaborative Filtering with Neighborhood-enriched Contrastive Learning".

119 stars 20 forks source link

Closed kaishuibujiatang closed 2 years ago

kaishuibujiatang commented 2 years ago

hyp1231 commented 2 years ago

Hi! 可以尝试将本项目下的 ncl.py 中的函数 forward, calculate_loss, full_sort_predict 替换为 RecBole 的对应模型的实现，例如： ngcf.py 和 dgcf.py。

kaishuibujiatang commented 2 years ago

好的，感谢感谢