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.

Sign up for GitHub

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

Jump to bottom

Node classification #12

Closed

steelman97 opened this issue Aug 5, 2021 · 1 comment

steelman97 commented Aug 5, 2021

What do I have to modify the code in order to try the model on ogbn-proteins? (Wrapper, collator..)

Contributor

chengxuanying commented Aug 8, 2021

Hi @steelman97 ,
I think writing a new dataset for node tasks is better. Please refer to the class MyGraphPropPredDataset if you want to do so.

There are some tips for trying the model on ogbn-proteins:

Implement a proper graph sampling strategy as mentioned in our paper.
Some parts of the code need to rewrite to fit for larger graph. (E.g., the floyd-warshall algorithm is n^3 regarding the number of nodes, you may need to choose an alternative metric for efficient spatial encoding.)

zhengsx closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment