Huhujingjing
/

custom-gcn

Graph Machine Learning

feature-extraction

Model card Files Files and versions Community

Huhujingjing commited on Jul 1, 2023

Commit

1b1e134

·

1 Parent(s): 7874c27

Upload model

Files changed (1) hide show

modeling_gcn.py +1 -1

modeling_gcn.py CHANGED Viewed

@@ -41,7 +41,7 @@ class GCNNet(torch.nn.Module):
         self.conv2 = GCNConv(hidden_size, 32)
         self.mlp = MLPReadout(32, num_classes)
-    def forward(self, data):
         x, edge_index, batch = data.x.long(), data.edge_index, data.batch
         x = self.embedding(x.reshape(-1))

         self.conv2 = GCNConv(hidden_size, 32)
         self.mlp = MLPReadout(32, num_classes)
+    def forward_features(self, data):
         x, edge_index, batch = data.x.long(), data.edge_index, data.batch
         x = self.embedding(x.reshape(-1))