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

Train and Test splits on other datasets #4

Closed
AmeenAli opened this issue Jun 30, 2021 · 3 comments
Closed

Train and Test splits on other datasets #4

AmeenAli opened this issue Jun 30, 2021 · 3 comments

Comments

@AmeenAli
Copy link

Hello,
I am trying to reproduce the results for the other datasets as well,
for Cora dataset for example, what is the train/test split you have used (this dataset is for node-classification), I am not aware of formal test/train split, did you take some percentage as train and the rest as test?
any clarification about this will help a lot
Thansk!

@shyam196
Copy link
Collaborator

shyam196 commented Jul 5, 2021

We used the default one provided by PyTorch geometric ("public"). See here: https://pytorch-geometric.readthedocs.io/en/latest/modules/datasets.html#torch_geometric.datasets.Planetoid

@shyam196 shyam196 closed this as completed Jul 5, 2021
@AmeenAli
Copy link
Author

AmeenAli commented Jul 6, 2021

@shyam196
Thanks for you reply,
I saw this written in your supps material at the end of the paper, however I am not aware of test/train splits for the 'public split', are you using cross-folds validation in case?
how can you load the test/train splits for the public split?
Best,

@shyam196
Copy link
Collaborator

shyam196 commented Jul 6, 2021

it's the split from this paper: https://arxiv.org/abs/1603.08861

this example from PyG's repo shows how to use it: https://github.com/rusty1s/pytorch_geometric/blob/master/examples/gcn2_cora.py

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