GNNExplainer `mask_features` option

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

Closed alirezadizaji closed 2 years ago

alirezadizaji commented 2 years ago

mask_features option is an argument of forward for GNNExplainer but it is not passed to gnn_explainer_alg. I actually have fixed and merged it to dig branch but not to dig-stable. https://github.com/divelab/DIG/blob/d2d648f7c5ab915ad64eff1c22660a84bba6d755/dig/xgraph/method/gnnexplainer.py#L160

Oceanusity commented 2 years ago

Thank you for pointing it out. I have added the parameter mask_features to this function.