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