f6532b3c4c329e6d5d5fb846acc441df47616c4c,torch_geometric/nn/models/gnn_explainer.py,GNNExplainer,explain_node,#GNNExplainer#Any#Any#Any#,51

Before Change

            optimizer.step()

        node_feat_masks = self.node_feat_masks
        node_feat_masks = [mask.detach().sigmoid() for mask in node_feat_masks]
        edge_masks = [mask.detach().sigmoid() for mask in self.edge_masks]

        self.__clear_masks__()

After Change

    def explain_node(self, node_idx, x, edge_index, **kwargs):
        self.model.eval()
        num_edges = edge_index.size(1)

        // Only operate on a k-hop subgraph around `node_idx`.
        x, edge_index, hard_edge_mask, kwargs = self.__subgraph__(
            node_idx, x, edge_index, **kwargs)

Instances