Skip to content

Conversation

@wsad1
Copy link
Member

@wsad1 wsad1 commented Mar 6, 2021

#2203.

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

@rusty1s
Copy link
Member

rusty1s commented Mar 7, 2021

Awesome! Thanks a lot :)

@rusty1s rusty1s merged commit b533f45 into pyg-team:master Mar 7, 2021
@wsad1 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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants