search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.68k stars 3.59k forks source link

Simply add the types of captum support #8773

Open lck-handsome opened 6 months ago

lck-handsome commented 6 months ago

🛠 Proposed Refactor

The algorithms supported by captum can be easily extended in the explain module

Suggest a potential alternative/fix

In captum_explainer.py, supported_methods=[] can be added to 'FeatureAblation', which is built into captum, without further modification to fit the current explain module

rusty1s commented 6 months ago

Thanks. Do you mind contributing this change?

lck-handsome commented 6 months ago

Of course it's my pleasure！