From 1ae4584e5b2e74d1e9ab7f64bf468373e5c7d960 Mon Sep 17 00:00:00 2001 From: Darryl Ong Date: Fri, 13 Oct 2023 16:51:43 +0800 Subject: [PATCH] Updated message handling --- cornac/models/lightgcn2/lightgcn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cornac/models/lightgcn2/lightgcn.py b/cornac/models/lightgcn2/lightgcn.py index 15cae5010..cb9ea1bc3 100644 --- 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