-
Notifications
You must be signed in to change notification settings - Fork 276
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
edge_mask=gnnexplainer_related_preds[1][prediction], NameError: name 'prediction' is not defined #240
Comments
Hi, the prediction results is missed, and should add codes Thanks for pointing it out. |
Or you can run the visualization for subgraphx first, and then you will have the prediction variable. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
See tutorials/KDD2022/xgraph_code_tutorial.ipynb
gnnexplainer_related_preds =
gnnexplainer(data.x, data.edge_index, sparsity=sparsity, num_classes=num_classes, node_idx=node_idx)
ax, G = gnnexplainer.visualize_graph(node_idx=node_idx,
edge_index=data.edge_index,
edge_mask=gnnexplainer_related_preds[1][prediction],
y=data.y)
And therefore there is no example of visualisation for GNNExplainer.
The text was updated successfully, but these errors were encountered: