-
Notifications
You must be signed in to change notification settings - Fork 209
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
Feature/gnn #501
base: master
Are you sure you want to change the base?
Feature/gnn #501
Conversation
…ccuracy on nodes, and 67% AUC for link prediction, even in this simple model. For Cody/Tanmoy when I am in France
i'm unclear what problem this solves? rgcn for link prediction that handles both node+edge attrs and node+edge classes? or for ER, how would this line up with a real dataset (hypergraph of ID -> attributes for raw input, and attribute<>attribute weighted equality scores of diff classes from feature engineering), + task (???) |
This gives them a base class that doesn't use DGL, is lightweight, and stubs the net while I am away. |
Adds joint training GNN toy model and fits it on Cora dataset.
For @tanmoyio and @webcoderz as templates for proj work.