graphembedding 0.1

Python Graph Embedding Library for Knowledge graph

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)