-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Add dropout_node
to torch_geometric.utils
#5481
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
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5481 +/- ##
==========================================
+ Coverage 83.30% 83.31% +0.01%
==========================================
Files 350 350
Lines 19049 19062 +13
==========================================
+ Hits 15868 15881 +13
Misses 3181 3181
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
Planning to add these methods in the following PRs:
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for adding this. Can we also create a small test for this?
Of course. Will add some tests after addressing your comments :) |
Co-authored-by: Matthias Fey <[email protected]>
Co-authored-by: Matthias Fey <[email protected]>
Co-authored-by: Matthias Fey <[email protected]>
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you!
@EdisonLeeeee Can you also add the changes to |
@rusty1s done. Will add |
* add dropout_node * pass num_nodes to subgraph * drop relabel_nodes argument * Update torch_geometric/utils/dropout.py Co-authored-by: Matthias Fey <[email protected]> * Update torch_geometric/utils/dropout.py Co-authored-by: Matthias Fey <[email protected]> * Update torch_geometric/utils/dropout.py Co-authored-by: Matthias Fey <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update * test * changelog Co-authored-by: Matthias Fey <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This PR is my first attempt to add graph augmentation methods into PyG. As discussed in #5452, I start with a simple augmentation method
DropNode
proposed in Graph Contrastive Learning with Augmentations.