Python Graph Embedding Libary for Knowledge graph
This project provides Tensorflow2.0 implementatinons of several different popular graph embeddings for knowledge graph.
graphembedding will be released on pypi soon.
python setup.py installIt'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)