LibKGE 是一个基于 PyTorch 的库,用于(KGE) 的高效训练、评估和超参数优化。 它具有高度可配置性、易于使用和可扩展性。 其他 KGE 框架。
LibKGE 的主要目标是促进对KGE 模型和训练方法的可重复研究(以及之间有意义的比较)。 正如我们在(请参见)中所述,训练策略和超参数的选择对模型性能的影响非常大,通常比模型类本身影响更大。 LibKGE 旨在提供可用于任何模型的训练、超参数优化和评估策略的干净实现。 框架中实现的每个潜在的旋钮或启发式方法都通过详细记录的配置文件显式公开(例如,请参阅和)。 LibKGE 还提供了最常见的 KGE 模型,并且可以轻松添加新模型(欢迎投稿!)。
对于链接预测任务, 等基于规则的系统是 KGE 的竞争替代品。
快速开始
# retrieve and install project in development mode
git
2022-03-21 20:37:23
416KB
Python
1