mage icon indicating copy to clipboard operation
mage copied to clipboard

Link prediction with DGL

Open as51340 opened this issue 2 years ago • 0 comments

Description

Implementing link prediction module with DGL. Work includes train, predict, set_model_parameters and test method. The goal is to implement a robust system that can work with a wide variety of datasets.

Pull request type

  • [ ] Bugfix
  • [ ] Algorithm/Module
  • [x] Feature
  • [ ] Code style update (formatting, renaming)
  • [ ] Refactoring (no functional changes, no api changes)
  • [ ] Build related changes
  • [ ] Documentation content changes
  • [ ] Other (please describe): ######################################

Reviewer checklist (the reviewer checks this part)

Module/Algorithm

  • [ ] Core algorithm/module implementation
  • [ ] Query module implementation
  • [ ] Unit tests
  • [ ] End-to-end tests
  • [ ] Code documentation
  • [ ] README short description
  • [ ] Documentation on memgraph/docs

######################################

as51340 avatar Aug 01 '22 12:08 as51340