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

[Model] SIGN for OGB dataset #2316

Merged
merged 6 commits into from
Nov 3, 2020
Merged

[Model] SIGN for OGB dataset #2316

merged 6 commits into from
Nov 3, 2020

Conversation

lingfanyu
Copy link
Collaborator

Description

SIGN implementation for node predictions on ogbn-products, ogbn-arxiv and ogbn-mag.

Accuracy (10 runs, Tesla T4)

Dataset Test Accuracy Validation Accuracy # Params
ogbn-products 0.8052±0.0016 0.9299±0.0004 3,483,703
ogbn-arxiv 0.7195±0.0011 0.7323±0.0006 3,566,128
ogbn-mag 0.4046±0.0012 0.4068±0.0010 3,724,645

Checklist

  • The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]])
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented
  • To the my best knowledge, examples are either not affected by this change,
    or have been fixed to be compatible with this change

- ogb 1.2.3


How to Rrun
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Run

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed. I also updated the readme in ogb folder

Dependencies
------------
- pytorch 1.5
- dgl 0.5 (tested with 0.6a201020 nightly build)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think nightly version is required for ogbn-mag due to the bug we discussed last time.

Copy link
Member

@mufeili mufeili left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@mufeili mufeili merged commit b8cc26e into dmlc:master Nov 3, 2020
BarclayII pushed a commit to BarclayII/dgl that referenced this pull request Nov 27, 2020
* sign for ogbn products, arxiv, mag

* texts

* fix

* update ogb folder readme

* use dgl nightly build

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
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

Successfully merging this pull request may close these issues.

None yet

2 participants