OpenKE:知识图谱表示学习工具包

OpenKE是THUNLP基于TensorFlow、PyTorch开发的用于将知识图谱嵌入到低维连续向量空间进行表示的开源框架。在OpenKE中,我们提供了快速且稳定的各类接口,也实现了诸多经典的知识表示学习模型。该框架易于扩展,基于框架设计新的知识表示模型也十分的方便。具体来说,OpenKE具有如下特点:

  1. 接口设计简单,可以轻松在各种不同的训练环境下部署模型。
  2. 底层的数据处理进行了优化,模型训练速度较快。
  3. 提供了轻量级的C++模型实现,在CPU多线程环境下也能快速运行。
  4. 提供了大规模知识图谱的预训练向量,可以直接在下游任务中使用。
  5. 长期的工程维护来解决问题和满足新的需求。

工具包地址:

https://github.com/thunlp/OpenKE