araison
/
eixgnn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
araison 2023-03-06 11:47:32 +01:00
parent 3109a36e6e
commit 79283e9a9d
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
eixgnn/eixgnn.py
View File

@ -265,7 +265,3 @@ class EiXGNN(ExplainerAlgorithm):
sub_pred = self.model(x=x, edge_index=edge_index_sub) sub_pred = self.model(x=x, edge_index=edge_index_sub)
node_importance[node_index] = torch.norm(pred - sub_pred, p=1) node_importance[node_index] = torch.norm(pred - sub_pred, p=1)
return node_importance.to(x.device) return node_importance.to(x.device)
if __name__ == "__main__":