Skip to content
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
dataloader
models
Readme.md
eval.py
kge_mxnet.ipynb
kge_mxnet_hypertune.ipynb
kge_pytorch.ipynb
kge_pytorch_hypertune.ipynb
train.py
train_mxnet.py
train_pytorch.py

Readme.md

Example of Knowledge Graph Embedding Using Deep Graph Library

This is an example of knowledge graph embedding (KGE) using Deep Graph Library (DGL). It uses the FB15K dataset, a knowledge base of general facts. An example use case would be to graph the relationships of persons and predict their nationality.

The example notebooks demonstrate implementations of Apache MXNet and PyTorch for the DGL backend. They use Amazon SageMaker deep learning containers that are preconfigured with DGL, so you don't have to build the dependencies yourself.

For more information about Deep Graph Library (DGL) please visit the DGL documentation website: https://docs.dgl.ai

You can’t perform that action at this time.