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

I have a list of graphs on which I want to perform node classification task. Each graph has nodes, node attributes, edges and associated labels for each node. Any guidelines on how to train this dataset? #18

Open
clydebailey opened this issue Apr 23, 2024 · 1 comment

Comments

@clydebailey
Copy link

No description provided.

@clydebailey clydebailey changed the title I have a list of graphs on which I want to perform node classification task. Any guidelines on how to create a dataset for such a dataset? I have a list of graphs on which I want to perform node classification task. Each graph has nodes, node attributes, edges and associated labels for each node. Any guidelines on how to train this dataset? Apr 23, 2024
@qitianwu
Copy link
Owner

Hi, we release an extended version of our follow-up work DIFFormer to accommodate a batch of graphs as input. The implementation allows for graphs with different sizes that are disconnected (the global attentions are computed within nodes in each graph). See the model implementation here https://github.com/qitianwu/DIFFormer/blob/extension/physical%20particle/difformer-v2.py

The extension for NodeFormer can be done in a similar way as DIFFormer.

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

No branches or pull requests

2 participants