Merge pull request #1900 from zhengbw0324/master

FIX: fix the note in LightGCN
RUCAIBox · Oct 31, 2023 · 27ea565 · 27ea565
2 parents 4caa9be + dcec484
commit 27ea565
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/model/general_recommender/lightgcn.py b/recbole/model/general_recommender/lightgcn.py
@@ -169,7 +169,7 @@ def calculate_loss(self, interaction):
         neg_scores = torch.mul(u_embeddings, neg_embeddings).sum(dim=1)
         mf_loss = self.mf_loss(pos_scores, neg_scores)
 
-        # calculate BPR Loss
+        # calculate regularization Loss
         u_ego_embeddings = self.user_embedding(user)
         pos_ego_embeddings = self.item_embedding(pos_item)
         neg_ego_embeddings = self.item_embedding(neg_item)