search

graphsinspace / graspe

Graphs in Space: Graph Embeddings for Machine Learning on Complex Data
GNU General Public License v3.0
7 stars 0 forks source link

GraphSAGE impl. #8

Closed nmilosev closed 3 years ago

nmilosev commented 3 years ago

Add new embedding.

nmilosev commented 3 years ago

Pytorch impl. https://github.com/williamleif/graphsage-simple (faster for smaller graphs)

Main impl. https://github.com/williamleif/GraphSAGE

https://pytorch-geometric.readthedocs.io/en/latest/modules/nn.html#torch_geometric.nn.conv.SAGEConv

https://github.com/rusty1s/pytorch_geometric/blob/master/examples/ogbn_products_sage.py

nmilosev commented 3 years ago

Implemented https://github.com/nmilosev/graspe/commit/389c20fad567c356003278e848e626470b35d603