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] GraphSAGE recommendation with custom distributed sampler #675

Closed
wants to merge 23 commits into from

Conversation

BarclayII
Copy link
Collaborator

@BarclayII BarclayII commented Jun 20, 2019

Description

GraphSAGE + custom distributed sampler on MovieLens dataset, both for rating prediction and link predictions.

This also includes a custom distributed sampler which sends auxiliary data along with NodeFlows.

As a result, two C APIs SerializeSubgraph and DeserializeSubgraph are added, which converts between C NodeFlow Handles and Python bytearray's.

Checklist

Please feel free to remove inapplicable items for your PR.

  • The PR title starts with [$CATEGORY] (such as [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

@jermainewang jermainewang mentioned this pull request Aug 2, 2019
22 tasks
@jermainewang
Copy link
Member

@yzh119 Is it possible to implement GraphSAGEConv in nn module first? This application could be a good test for the module.

@BarclayII BarclayII closed this May 5, 2020
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