speed up gnn explainer on node-level

divelab / DIG

A library for graph deep learning research

https://diveintographs.readthedocs.io/

GNU General Public License v3.0

1.84k stars 281 forks source link

speed up gnn explainer on node-level #113

Closed alirezadizaji closed 2 years ago

alirezadizaji commented 2 years ago

Hi, in node-level explanation for GNNExplainer, we could only take nodes and edges of its K-Hop subset for optimization in order to speed up the procedure (from 17.23s to 1.85s with CPU)