From 91ed98c64f358f925d0a183ec8053a62e7fb7b0a Mon Sep 17 00:00:00 2001 From: Theis Jendal <tjendal@cs.aau.dk> Date: Mon, 25 Mar 2024 16:52:02 +0100 Subject: [PATCH] Fix layer combination coefficients (#606) --- cornac/models/lightgcn/lightgcn.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/cornac/models/lightgcn/lightgcn.py b/cornac/models/lightgcn/lightgcn.py index c2f0dba9..fbe37a46 100644 --- a/cornac/models/lightgcn/lightgcn.py +++ b/cornac/models/lightgcn/lightgcn.py @@ -96,9 +96,6 @@ class Model(nn.Module): item_embeds = h_dict[ITEM_KEY] iterator = enumerate(zip([in_g] * len(self.layers), self.layers)) - user_embeds = user_embeds * (1 / (len(self.layers) + 1)) - item_embeds = item_embeds * (1 / (len(self.layers) + 1)) - for k, (g, layer) in iterator: h_dict = layer(g, h_dict) ue = h_dict[USER_KEY] -- GitLab