KGATE : a Knowledge Graph Embedding Training Environment

我们介绍了KGATE（知识图谱自编码器训练环境），一个基于PyTorch Geometric和TorchKGE构建的模块化Python库。KGATE允许用户将初始器、编码器、解码器、损失函数、负采样器和评估指标作为构建块组装起来，或自行添加其他模块。KGATE包含控制数据泄露的预处理流程、内置训练管道，并通过设计实现可重复性。