Updated message handling

PreferredAI · Oct 13, 2023 · 1ae4584 · 1ae4584
1 parent 8797bc5
commit 1ae4584
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cornac/models/lightgcn2/lightgcn.py b/cornac/models/lightgcn2/lightgcn.py
@@ -45,7 +45,7 @@ def forward(self, g, feat_dict):
             src, dst = g.edges(etype=(srctype, etype, dsttype))
             norm = self.norm_dict[(srctype, etype, dsttype)]
             # TODO: CHECK HERE
-            messages = norm * feat_dict[srctype][src] * feat_dict[dsttype][dst]  # compute messages
+            messages = norm * feat_dict[srctype][src]  # compute messages
             g.edges[(srctype, etype, dsttype)].data[
                 etype
             ] = messages  # store in edata