def attention_scores()

in src/model.py [0:0]

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

    def attention_scores(self, g: dgl.DGLGraph) -> torch.Tensor:
        """Calculate attention scores"""
        h = self._conv_forward(g)
        with g.local_scope():
            gate = self.gap.gate_nn(h)
            g.ndata["gate"] = gate
            gate = dgl.softmax_nodes(g, "gate")
            g.ndata.pop("gate")
            return gate