From dcec4848ca7977fb4b77c642ceef85c3164e96c3 Mon Sep 17 00:00:00 2001 From: zbw <18735382001@163.com> Date: Tue, 31 Oct 2023 14:52:04 +0800 Subject: [PATCH] FIX: fix the note in LightGCN --- recbole/model/general_recommender/lightgcn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recbole/model/general_recommender/lightgcn.py b/recbole/model/general_recommender/lightgcn.py index 36308b9dd..de9d9dc14 100644 --- 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)