pykeen

A Python library for learning and evaluating knowledge graph embeddings

code:example.py

from pykeen.pipeline import pipeline

result = pipeline(model='TransE', dataset='nations')