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.

Sign up for GitHub

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

Jump to bottom

Added a flag to gnn_explainer. #2209

Merged

rusty1s merged 2 commits into pyg-team:master from wsad1:explainerflag

Mar 7, 2021

Member

wsad1 commented Mar 6, 2021 •

edited

Loading

Made changes to.

gnn_explainer.py
- Added a flag called retrun_type to gnn_explainer. This flag indicates if the model will output raw scores, probabilities or log of probability.
- Within the explian_node module model outputs are converted to log of probability based on the flag. to_log_proba is used to convert outputs.
test_gnn_explainer.py: Add a new test function to test to_log_proba

wsad1 and others added 2 commits

March 6, 2021 17:52


          added a flag to gnn_explainer

acb905b


          cleanup

3a96f2f

Member

rusty1s commented Mar 7, 2021

Awesome! Thanks a lot :)

rusty1s merged commit b533f45 into pyg-team:master

wsad1 deleted the explainerflag branch

March 27, 2021 13:02

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment