Skip to content

Commit 27ea565

Browse files
authored
Merge pull request #1900 from zhengbw0324/master
FIX: fix the note in LightGCN
2 parents 4caa9be + dcec484 commit 27ea565

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recbole/model/general_recommender/lightgcn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def calculate_loss(self, interaction):
169169
neg_scores = torch.mul(u_embeddings, neg_embeddings).sum(dim=1)
170170
mf_loss = self.mf_loss(pos_scores, neg_scores)
171171

172-
# calculate BPR Loss
172+
# calculate regularization Loss
173173
u_ego_embeddings = self.user_embedding(user)
174174
pos_ego_embeddings = self.item_embedding(pos_item)
175175
neg_ego_embeddings = self.item_embedding(neg_item)

0 commit comments

Comments
 (0)