def _conv_forward()

in src/model.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def _conv_forward(self, g: dgl.DGLGraph) -> torch.Tensor:
        """forward pass through the GraphConv layers"""
        in_feat = g.ndata["h"]
        h = self.conv1(g, in_feat)
        h = F.relu(h)
        h = self.conv2(g, h)
        h = F.relu(h)
        return h