Back to Pytorch Geometric

Examples for Generating Explanations of Graph Neural Networks

examples/explain/README.md

2.7.01.3 KB
Original Source

Examples for Generating Explanations of Graph Neural Networks

This directory contains examples demonstrating the use of the torch_geometric.explain package. The explain package of PyG provides a set of tools to explain the predictions of a GNN model or to explain the underlying phenomenon of a dataset.

ExampleDescription
gnn_explainer.pyGNNExplainer for node classification
gnn_explainer_link_pred.pyGNNExplainer for link prediction
gnn_explainer_ba_shapes.pyGNNExplainer applied on the BAShapes dataset
captum_explainer.pyCaptum-based explainer for node classification
captum_explainer_hetero_link.pyCaptum-based explainer for heterogenous link prediction
graphmask_explainer.pyGraphMaskExplainer for node classification