Skip to content
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

Explainer: Support for setting and clearing masks in heterogeneous GNNs #5886

Merged
merged 11 commits into from
Nov 4, 2022

Conversation

wsad1
Copy link
Member

@wsad1 wsad1 commented Nov 3, 2022

#5630 First step in support to_captum for heterogenous graphs.

@wsad1 wsad1 requested a review from RBendias November 3, 2022 13:56
@wsad1 wsad1 self-assigned this Nov 3, 2022
@wsad1 wsad1 added the explain label Nov 3, 2022
@github-actions github-actions bot added the nn label Nov 3, 2022
@codecov
Copy link

codecov bot commented Nov 3, 2022

Codecov Report

Merging #5886 (bc5a620) into master (0bdd24c) will increase coverage by 0.00%.
The diff coverage is 100.00%.

❗ Current head bc5a620 differs from pull request most recent head ac76a7a. Consider uploading reports for the commit ac76a7a to get more accurate results

@@           Coverage Diff           @@
##           master    #5886   +/-   ##
=======================================
  Coverage   84.43%   84.44%           
=======================================
  Files         349      349           
  Lines       19491    19499    +8     
=======================================
+ Hits        16457    16465    +8     
  Misses       3034     3034           
Impacted Files Coverage Δ
torch_geometric/nn/models/explainer.py 96.07% <100.00%> (+0.21%) ⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@rusty1s rusty1s changed the title Explainer: Support setting and clearing masks for HeteroConv modules. Explainer: Support setting and clearing masks for HeteroConv modules Nov 4, 2022
@rusty1s rusty1s self-assigned this Nov 4, 2022
@rusty1s rusty1s marked this pull request as ready for review November 4, 2022 07:41
Copy link
Member

@rusty1s rusty1s left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @wsad1. Also added support for to_hetero().

@rusty1s rusty1s enabled auto-merge (squash) November 4, 2022 07:43
@rusty1s rusty1s changed the title Explainer: Support setting and clearing masks for HeteroConv modules Explainer: Support setting and clearing masks for heterogeneous GNNs Nov 4, 2022
@rusty1s rusty1s changed the title Explainer: Support setting and clearing masks for heterogeneous GNNs Explainer: Support for setting and clearing masks in heterogeneous GNNs Nov 4, 2022
@rusty1s rusty1s disabled auto-merge November 4, 2022 07:44
@rusty1s rusty1s enabled auto-merge (squash) November 4, 2022 07:44
@rusty1s rusty1s disabled auto-merge November 4, 2022 07:44
@rusty1s rusty1s enabled auto-merge (squash) November 4, 2022 07:45
@rusty1s rusty1s merged commit e832652 into master Nov 4, 2022
@rusty1s rusty1s deleted the hetero_masks branch November 4, 2022 07:46
JakubPietrakIntel pushed a commit to JakubPietrakIntel/pytorch_geometric that referenced this pull request Nov 25, 2022
…GNNs (pyg-team#5886)

pyg-team#5630 First step in
support `to_captum` for heterogenous graphs.

Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants