Introduce LambdaKG


We present LambdaKG, a library for KGE that equips with many pre-trained language models (e.g., BERT, BART, T5, GPT-3), and supports various tasks (e.g., knowledge graph completion, question answering, recommendation, and knowledge probing).


Learn More

280

Star

30

Fork

9

Models

6

Support Datasets

Features & architecture of LambdaKG

LambdaKG contains: 1) Core Module with Unified KG Encoder: LambdaKG utilizes a unified encoder to pack graph structure and text semantics, with convenient Trainer&Evaluator, Metric, and Bag of Tricks; 2) Model Hub: LambdaKG is integrated with many cutting-edge PLM-based KGE models; 3) Flexible Downstream Tasks: LambdaKG disentangles KG representation learning and downstream tasks.

Model Hubs

LambdaKG consists of a Model Hub which supports many representative language model-based KGE models.

Support LLMs based on APIs

We support LLMs for KG link prediction with OpenAI's API.

  • ChatGPT
  • GPT-3
Learn More