diff --git a/cornac/models/lightgcn/lightgcn.py b/cornac/models/lightgcn/lightgcn.py index c2f0dba92e429f13c72e43700d3a1e3bd9de7fe4..fbe37a460108fb7fdb48affef04aeb34126aa1c8 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]