plot_er

plot_er(pipeline_result, *, model: str | None = None, entities: set[str] | None = None, relations: set[str] | None = None, apply_limits: bool = True, margin: float = 0.4, plot_entities: bool = True, plot_relations: bool | None = None, annotation_x_offset: float = 0.02, annotation_y_offset: float = 0.03, entity_embedding_getter=None, relation_embedding_getter=None, ax=None, subtitle: str | None = None, **kwargs)[源代码]

在2D中绘制简化的实体和关系向量。

Parameters:

  • pipeline_result – 由pykeen.pipeline.pipeline()返回的结果。

  • model (str | None) – 来自 sklearn 的降维模型。默认为 PCA。 也可以使用 KPCA、GRP、SRP、TSNE、LLE、ISOMAP、MDS 或 SE。

  • 实体 (set[str] | None) – 要绘制的实体子集

  • relations (set[str] | None) – 要绘制的关系的子集

  • apply_limits (bool) – 是否应该应用x和y的限制？

  • margin (float) – 最小/最大x和y值周围的边距大小

  • plot_entities (bool) – 如果为真，根据其降维后的嵌入绘制实体

  • plot_relations (bool | None) – 默认情况下，这仅在平移距离模型上启用，例如 pykeen.models.TransE。

  • annotation_x_offset (float) – 标签相对于实体位置的X偏移量

  • annotation_y_offset (float) – 标签相对于实体位置的Y偏移量

  • entity_embedding_getter – 一个函数，它接受一个模型并返回其实体嵌入。如果没有提供，默认使用_default_entity_embedding_getter()，它只是获取model.entity_embeddings。注意，默认值仅适用于旧式的PyKEEN模型。

  • relation_embedding_getter – 一个函数，它接受一个模型并返回其关系嵌入。如果没有提供，默认使用 _default_relation_embedding_getter()，它只是获取 model.relation_embeddings。注意，默认值仅适用于旧式的 PyKEEN 模型。

  • ax – 如果预定义了，则为matplotlib的轴

  • 副标题 (str | None) – 用户定义的副标题。如果未提供，则会自动推断。传递空字符串以不使用副标题。

  • kwargs – 传递给reducer类（例如PCA, TSNE）的__init__()的关键字参数

Returns:

轴

Raises:

ValueError – 如果实体绘图和关系绘图都被关闭

警告

在同一图表上绘制关系和实体仅对像TransE这样的平移距离模型有意义。