pykeen
PyKEEN是一个专为知识图谱嵌入设计的Python开源库,支持多模态信息的训练与评估。通过pipeline函数提供高层次的可扩展功能,可以轻松训练和评估模型。内置37个数据集和多个模型,支持自定义数据集和模型扩展。集成了Optuna和PyTorch Lightning,适用于多种训练循环和评估方法。访问https://pykeen.readthedocs.io了解更多信息。