-
Notifications
You must be signed in to change notification settings - Fork 3.7k
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
Custom motif generation #6179
Custom motif generation #6179
Conversation
Codecov Report
@@ Coverage Diff @@
## master #6179 +/- ##
=======================================
Coverage 84.55% 84.55%
=======================================
Files 372 372
Lines 20848 20848
=======================================
Hits 17629 17629
Misses 3219 3219 📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
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 for his @cuent, I've left some minor comment.
This PR implements benchmark dataset overall framework and API for generating benchmark datasets (#5817). Any feedback is appreciated, especially regarding the proposed architecture. ### Implementing a new graph generator - Create a class inside generators inheriting from GraphGenerator - Implement `generate_base_graph` using the provided methods (`generate_feature` and `attach_motif`) - See #6072 and #6073 to check how it was implemented - The current Motif generator used here will be replaced by #6179 ### Example for the final user ``` motif = Motif('house') generator = BAGraph(num_nodes=300, num_motifs=80, motif=motif) dataset = ExplainerDataset(generator) ``` TODO: - [x] Finish graph generator (provide all necessary methods: feature generator, label generator) - [x] Add tests - [x] Add to Change log - [x] Documentation Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>
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! Aligned the code to the new interface :)
This task is part of GNN Explainability Dataset Generation
Implementation of the base class for motif generation. It depends on the framework and API to generate benchmark datasets.
The base class follows some of the structure from @rfdavid PRs. However, I changed a bit since I believe using Data is a better and cleaner approach to generate structures and create wrappers for other structures in PyG or NetworkX.
Once I have some feedback about the
MotifGenerator
, I will go ahead andGraphGenerator.attach_motif()