mage
mage copied to clipboard
Link prediction with DGL
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
######################################