Fix: inter_feature is self-quadratic

zyy598 · Dec 14, 2019 · dab66a1 · dab66a1
1 parent 1853bd4
commit dab66a1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/GraphNCF/GCFmodel.py b/GraphNCF/GCFmodel.py
@@ -76,7 +76,8 @@ def forward(self, laplacianMat,selfLoop,features):
         L1 = laplacianMat + selfLoop
         L2 = laplacianMat.cuda()
         L1 = L1.cuda()
-        inter_feature = torch.mul(features,features)
+        inter_feature = torch.sparse.mm(L2,features)
+        inter_feature = torch.mul(inter_feature,features)
 
         inter_part1 = self.linear(torch.sparse.mm(L1,features))
         inter_part2 = self.interActTransform(torch.sparse.mm(L2,inter_feature))
@@ -179,4 +180,4 @@ def forward(self,userIdx,itemIdx):
 
     rt['userId'] = rt['userId'] - 1
     rt['itemId'] = rt['itemId'] - 1
-    gcf = GCF(userNum,itemNum,rt)
+    gcf = GCF(userNum,itemNum,rt)