-
Notifications
You must be signed in to change notification settings - Fork 3k
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
[GraphBolt][Dataset] Contribute IGBH dataset to hetero examples. #7708
base: master
Are you sure you want to change the base?
Conversation
To trigger regression tests:
|
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
examples/graphbolt/rgcn/evaluator.py
Outdated
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.
We don't need this file. The same functionality can be implemented in a few lines.
"igb-het-medium", | ||
"igb-het-large", | ||
"igb-het", | ||
"igb-het-MLPerf", |
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.
Can we make it all lower case?
The igb-hom-[tiny|small|medium] dataset is a heterogeneous citation network, | ||
which is designed for developers to train and evaluate GNN models with | ||
high fidelity. See more details in `igb-het-[tiny|small|medium] | ||
high fidelity. See more details in `igb-het-[tiny|small|medium|large] |
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.
also here as well. igb-het-[tiny|small|medium|large|mlperf]
and igb-het
.
LGTM except for the overly complicated evaluator.py file. |
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.
Let's use pure pytorch operations to compute the evaluation score.
Description
I added IGB-dataset to the RGCN folder. Currently the three smaller version (tiny[~2GB], small[~20GB], medium[~100GB]) are supported. The two larger versions (large[~800GB], full[~2.2TB]) are still under test.
Usage:
Checklist
Please feel free to remove inapplicable items for your PR.
Changes