A scalable Gensim implementation of "Learning Role-based Graph Embeddings" (IJCAI 2018).

Project Name | Stars | Downloads | Repos Using This | Packages Using This | Most Recent Commit | Total Releases | Latest Release | Open Issues | License | Language |
---|---|---|---|---|---|---|---|---|---|---|

Deepwalk | 2,513 | 7 | 5 months ago | 4 | April 29, 2018 | 42 | other | Python | ||

DeepWalk - Deep Learning for Graphs | ||||||||||

Gemsec | 234 | 7 months ago | gpl-3.0 | Python | ||||||

The TensorFlow reference implementation of 'GEMSEC: Graph Embedding with Self Clustering' (ASONAM 2019). | ||||||||||

Splitter | 200 | 7 months ago | 1 | gpl-3.0 | Python | |||||

A Pytorch implementation of "Splitter: Learning Node Representations that Capture Multiple Social Contexts" (WWW 2019). | ||||||||||

Role2vec | 157 | 7 months ago | gpl-3.0 | Python | ||||||

A scalable Gensim implementation of "Learning Role-based Graph Embeddings" (IJCAI 2018). | ||||||||||

Diff2vec | 116 | 7 months ago | gpl-3.0 | Python | ||||||

Reference implementation of Diffusion2Vec (Complenet 2018) built on Gensim and NetworkX. | ||||||||||

Rolx | 56 | 7 months ago | gpl-3.0 | Python | ||||||

An alternative implementation of Recursive Feature and Role Extraction (KDD11 & KDD12) | ||||||||||

Grarep | 54 | 7 months ago | 1 | gpl-3.0 | Python | |||||

A SciPy implementation of "GraRep: Learning Graph Representations with Global Structural Information" (WWW 2015). | ||||||||||

Subgraph2vec_gensim | 36 | 6 years ago | 1 | Python | ||||||

Contains the code (and working vm setup) for our KDD MLG 2016 paper titled: "subgraph2vec: Learning Distributed Representations of Rooted Sub-graphs from Large Graphs" | ||||||||||

Graph_representation_learning | 14 | 3 years ago | 1 | Python | ||||||

Nbne | 11 | 4 years ago | mit | Python | ||||||

Code to use Neighbor Based Node Embeddings (NBNE) method to create representations to nodes in a graph. |

Alternatives To Role2vecSelect To Compare

Alternative Project Comparisons

Readme

A scalable parallel **gensim** implementation of **Learning Role-based Graph Embeddings (IJCAI 2018)**.

Random walks are at the heart of many existing network embedding methods. However, such algorithms have many limitations that arise from the use of random walks, e.g., the features resulting from these methods are unable to transfer to new nodes and graphs as they are tied to vertex identity. In this work, we introduce the Role2Vec framework which uses the flexible notion of attributed random walks, and serves as a basis for generalizing existing methods such as DeepWalk, node2vec, and many others that leverage random walks. Our proposed framework enables these methods to be more widely applicable for both transductive and inductive learning as well as for use on graphs with attributes (if available). This is achieved by learning functions that generalize to new nodes and graphs. We show that our proposed framework is effective with an average AUC improvement of 16.55% while requiring on average 853x less space than existing methods on a variety of graphs.

The second-order random walk sampling methods were taken from the reference implementation of Node2vec.

The model is now also available in the package Karate Club.

This repository provides an implementation of Role2Vec as described in the paper:

Learning Role-based Graph Embeddings. Nesreen K. Ahmed, Ryan Rossi, John Boaz Lee, Theodore L. Willke, Rong Zhou, Xiangnan Kong, Hoda Eldardiry. StarAI workshop - IJCAI, 2018. [Paper]

The codebase is implemented in Python 3.5.2. package versions used for development are just below.

```
networkx 2.4
tqdm 4.28.1
numpy 1.15.4
pandas 0.23.4
texttable 1.5.0
scipy 1.1.0
argparse 1.1.0
gensim 3.6.0
scikit-learn 0.20.0
```

The code takes the **edge list** of the graph in a csv file. Every row indicates an edge between two nodes separated by a comma. The first row is a header. Nodes should be indexed starting with 0. A sample graph for `Cora` is included in the `input/` directory.

Training the model is handled by the `src/main.py` script which provides the following command line arguments.

```
--graph-input STR Input graph path. Default is `input/cora_edges.csv`.
--output STR Embeddings path. Default is `output/cora_role2vec.csv`.
```

```
--window-size INT Skip-gram window size. Default is 5.
--walk-number INT Number of walks per node. Default is 10.
--walk-length INT Number of nodes in walk. Default is 80.
--sampling STR Sampling procedure. Default is `first`.
--P FLOAT Return parameter. Default is 1.0.
--Q FLOAT In-out parameter. Default is 1.0.
```

```
--dimensions INT Number of dimensions. Default is 128
--down-sampling FLOAT Down sampling frequency. Default is 0.001.
--alpha FLOAT Initial learning rate. Default is 0.025.
--min-alpha FLOAT Final learning rate. Default is 0.025.
--min-count INT Minimal feature count. Default is 1
--workers INT Number of cores. Default is 4.
--epochs INT Number of epochs. Default is 10.
```

```
--features STR Feature extraction mechanism. Default is `wl`.
--labeling-iterations INT Number of WL labeling iterations. Default is 2.
--log-base FLOAT Log base for label creation. Default is 1.5.
--graphlet-size INT Maximal graphlet size. Default is 4.
--quantiles INT Number of quantiles for binning. Default is 5.
--motif-compression STR Motif compression procedure. Default is `string`.
--seed INT Sklearn random seed. Default is 42.
--factors INT Factors for motif compression. Default is 8.
--clusters INT Number of motif based labels. Default is 50.
--beta FLOAT Motif compression regularizer. Default is 0.01.
```

The following commands learn a Role2Vec embedding. The first example trains an embedding based on the default dataset with standard hyperparameter settings. The script saves the embedding at the default path.

``` python src/main.py ```

Using the degree centrality as a structural feature.

```
python src/main.py --features degree
```

Using the Weisfeiler-Lehman labeling as a structural feature.

```
python src/main.py --features wl
```

Using motif based structural features with factorization compression.

```
python src/main.py --features motif --motif-compression factorization
```

Using motif based structural features with factorization compression and a structural label number of 40.

```
python src/main.py --features motif --motif-compression factorization --clusters 40
```

Using a custom factorization dimension for the embedding.

```
python src/main.py --dimensions 32
```

Using second-order attributed ranom walks for sampling.

```
python src/main.py --sampling second --P 1 --Q 4
```

**License**

Popular Graph Projects

Popular Gensim Projects

Popular Computer Science Categories

Related Searches

Get A Weekly Email With Trending Projects For These Categories

No Spam. Unsubscribe easily at any time.

Python

Machine Learning

Deep Learning

Graph

Pytorch

Tensorflow

Random

Sklearn

Word2vec

Gensim

Representation Learning