update DHCF

iMoonLab · Dec 27, 2022 · 7293527 · 7293527
1 parent 2b25d32
commit 7293527
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/dhg/models/hypergraphs/dhcf.py b/dhg/models/hypergraphs/dhcf.py
@@ -77,7 +77,7 @@ def forward(
             u_embs = hg_ui.smoothing_with_HGNN(u_embs)
             i_embs = hg_iu.smoothing_with_HGNN(i_embs)
             g_embs = torch.cat([u_embs, i_embs], dim=0)
-            sum_embs = F.leaky_relu(self.W_gc[_idx](g_embs) + g_embs, negative_slope=0.2)
+            sum_embs = F.leaky_relu(self.W_gc[_idx](g_embs) + all_embs, negative_slope=0.2)
             #==========================================================================================
 
             bi_embs = all_embs * g_embs

diff --git a/dhg/nn/convs/hypergraphs/dhcf_conv.py b/dhg/nn/convs/hypergraphs/dhcf_conv.py
@@ -46,10 +46,10 @@ def forward(self, X: torch.Tensor, hg: Hypergraph) -> torch.Tensor:
             X (``torch.Tensor``): Input vertex feature matrix. Size :math:`(N, C_{in})`.
             hg (``dhg.Hypergraph``): The hypergraph structure that contains :math:`N` vertices.
         """
-        X = self.theta(X)
+        X_ = self.theta(X)
         if self.bn is not None:
-            X = self.bn(X)
-        X = hg.smoothing_with_HGNN(X) + X
+            X_ = self.bn(X_)
+        X_ = hg.smoothing_with_HGNN(X_) + X
         if not self.is_last:
-            X = self.drop(self.act(X))
-        return X
+            X_ = self.drop(self.act(X_))
+        return X_