Update README.md
This commit is contained in:
parent
3109a36e6e
commit
79283e9a9d
@ -265,7 +265,3 @@ class EiXGNN(ExplainerAlgorithm):
|
||||
sub_pred = self.model(x=x, edge_index=edge_index_sub)
|
||||
node_importance[node_index] = torch.norm(pred - sub_pred, p=1)
|
||||
return node_importance.to(x.device)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user