vlukiyanov / pt-splitter

PyTorch implementation of Splitter graph node embeddings

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pt-splitter

Build Status codecov Codacy Badge

Currently this is work in progress, it doesn't reproduce anything and is lacking more tests and further examples.

PyTorch implementation of a version of the Splitter algorithm for graph embeddings. Compatible with PyTorch 1.0.0 and Python 3.7 with or without CUDA.

This follows (or attempts to; note this implementation is unofficial) the algorithm described in "Is a Single Embedding Enough? Learning Node Representations that Capture Multiple Social Contexts" of Alessandro Epasto, Bryan Perozzi (http://epasto.org/papers/www2019splitter.pdf; the official implementation can be found at https://github.com/google-research/google-research/tree/master/graph_embedding/persona).

Examples

TODO

Usage

TODO

Other implementations of Splitter or similar

About

PyTorch implementation of Splitter graph node embeddings

License:MIT License


Languages

Language:Python 100.0%