Fix paper reference

PreferredAI · Oct 17, 2023 · 3992661 · 3992661
1 parent 0c81202
commit 3992661
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cornac/models/lightgcn/lightgcn.py b/cornac/models/lightgcn/lightgcn.py
@@ -1,5 +1,3 @@
-# Reference: https://github.com/dmlc/dgl/blob/master/examples/pytorch/NGCF/NGCF/model.py
-
 import torch
 import torch.nn as nn
 import torch.nn.functional as F

diff --git a/cornac/models/lightgcn/recom_lightgcn.py b/cornac/models/lightgcn/recom_lightgcn.py
@@ -69,7 +69,9 @@ class LightGCN(Recommender):
 
     References
     ----------
-    *   Wang, Xiang, et al. "Neural graph collaborative filtering." Proceedings of the 42nd international ACM SIGIR conference on Research and development in Information Retrieval. 2019.
+    *   He, X., Deng, K., Wang, X., Li, Y., Zhang, Y., & Wang, M. (2020).
+        LightGCN: Simplifying and Powering Graph Convolution Network for
+        Recommendation.
     """
 
     def __init__(