PyKEEN

PyKEEN 是一个用于可重复、简便的知识图谱嵌入的 Python 包。

最快上手的方法是使用pykeen.pipeline.pipeline()函数。

它提供了一个高级入口，用于扩展此包的功能。以下示例展示了如何通过名称引用在Nations数据集（pykeen.datasets.Nations）上训练和评估TransE模型（pykeen.models.TransE）。默认情况下，训练循环使用随机封闭世界假设训练方法（pykeen.training.SLCWATrainingLoop），并通过基于排名的评估方法进行评估（pykeen.evaluation.RankBasedEvaluator）。

>>> from pykeen.pipeline import pipeline
>>> result = pipeline(
...     model='TransE',
...     dataset='Nations',
... )

结果以pykeen.pipeline.PipelineResult实例的形式返回，该实例具有训练模型、训练循环和评估的属性。

PyKEEN 有一个函数 pykeen.env()，它可以神奇地打印出有关 PyTorch、CUDA 和您的操作系统的相关版本信息，这些信息可以用于调试。 如果您在 Jupyter 笔记本中，它将以 HTML 表格的形式漂亮地打印出来。

>>> import pykeen
>>> pykeen.env()

入门指南

参考

索引和表格