mage
mage copied to clipboard
MemTorch module
Description
Memgraph meets the ML! Integration of PyTorch Geometric. The idea is to have an off-the-shelf Pytorch data loader from Memgraph. User would have to write his own custom model to train and evaluate. The work is in progress.
Pull request type
- [x] Algorithm/Module
######################################
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
######################################