An Open-source library for Knowledge Embedding forked from github.com/thunlp/OpenKE. The original API changed drastically to be more pythonic.

For a full documentation see pyke.readthedocs.io.

This is an implementation based on TensorFlow for knowledge representation learning (KRL). It includes native C++ implementations for underlying operations such as data preprocessing and negative sampling. For each specific model, it is implemented by TensorFlow with Python interfaces so that there is a convenient platform to run models on GPUs.

1. Clone repository and enter directory

   git clone https://github.com/ifis-tu-bs/pyKE.git
   cd pyKE
   

2. Install package

   python setup.py install

To compute a knowledge graph embedding, first import datasets and set configure parameters for training, then train models and export results. Here is an example to train the FB15K dataset with the TransE model.

Please note that the dataset (fb15k.nt) is not included in the repository.

from pyke.dataset import Dataset
from pyke.embedding import Embedding
from pyke.models import TransE

# Read the dataset
dataset = Dataset("./fb15k.nt") 
embedding = Embedding(
    dataset,
    TransE,
    folds=20,
    epochs=20,
    neg_ent=1,
    neg_rel=0,
    bern=False,
    workers=4,
    dimension=50,  # TransE-specific
    margin=1.0,  # TransE-specific
)

# Train the model. It is saved in the process.
embedding.train(prefix="./TransE")

# Save the embedding to a JSON file
embedding.save_to_json("TransE.json")

The class pyke.embedding.Embedding represents an embedding which requires a dataset and a model class. Initialize your data set in form of a N-triples file with the class pyke.dataset.Dataset.

The class pyke.models.Model.Model declares the methods that all implemented model classes share, including the loss function neccessairy for training (inserting information into the model) and prediction (aka. retrieving information from the model). This project implements the following model classes:

• Analogy
• ComplEx
• DistMult
• HolE
• RESCAL
• TransD
• TransE
• TransH
• TransR

The original fork consists of a C++ library which is compiled once you use the project. Please note that the compilation is only supported on UNIX-based systems. Maybe this is changed in future versions.