Update README.md

This commit is contained in:
araison 2023-03-06 11:47:32 +01:00
parent 3109a36e6e
commit 79283e9a9d

View File

@ -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__":