erickguan / kgegrok

An efficient KGE learning framework

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Knowledge Graph Embedding Learning

PyPI version fury.io PyPI license

This is an learning modules with PyTorch to learn knowledge embeddings. It's more towards research now. It heavily used another library kgekit.

The design follows DIY, data oriented design and a few OO classes.

Features

  • Small memory footprint
  • Fast parallelized sampling and batch done in C++
  • Deterministic
  • Easily extensible and readable
  • Visualization

About

An efficient KGE learning framework


Languages

Language:Python 99.7%Language:Shell 0.3%