Change Edge Gradients to be not absolute

m30m · Feb 1, 2021 · ee81463 · ee81463
1 parent c8bbc01
commit ee81463
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/explain_methods.py b/explain_methods.py
@@ -154,7 +154,7 @@ def explain_sa(model, node_idx, x, edge_index, target, include_edges=None):
     saliency = Saliency(model_forward)
     input_mask = torch.ones(edge_index.shape[1]).requires_grad_(True).to(device)
     saliency_mask = saliency.attribute(input_mask, target=target,
-                                       additional_forward_args=(model, node_idx, x, edge_index))
+                                       additional_forward_args=(model, node_idx, x, edge_index), abs=False)
 
     edge_mask = saliency_mask.cpu().numpy()
     return edge_mask