GraphEmbedding

Python Graph Embedding Libary for Knowledge graph

This project provides Tensorflow2.0 implementatinons of several different popular graph embeddings for knowledge graph.

• transE
• complEx

Installation:

graphembedding will be released on pypi soon.

python setup.py install

Basic Usages:

It's simple. example code is below. The embedding object is returned as pd.Dataframe, so it can be used easily.

from graphembedding.playground import load_github
from graphembedding import complEx, transE

# Load Sample dataset 
github_dataset = load_github() 
triplets = github_dataset[['subject','relation','object']].values

# That's all. One line code.
node_embedding, edge_embedding = complEx(triplets) 

# if you wanna use transE,
# node_embedding, edge_embedding = transE(triplets)