predict_triples

predict_triples(model: Model, *, triples: None | Tensor | ndarray | tuple[str, str, str] | Sequence[tuple[str, str, str]], triples_factory: CoreTriplesFactory | None = None, batch_size: int | None = None, mode: Literal['training', 'validation', 'testing'] | None = None) → ScorePack[source]

对标记或映射的三元组进行预测。

Parameters:

  • 模型 (Model) – 模型。

  • triples (None | Tensor | ndarray | tuple[str, str, str] | Sequence[tuple[str, str, str]]) –

    形状: (num_triples, 3) 以下格式之一的三元组：

    • 一个基于标签的三元组。

    • 一个基于标签的三元组列表。

    • 一个基于标签的三元组数组。

    • 一个基于ID的三元组数组。

    • 无。在这种情况下，必须提供一个三元组工厂，并使用其三元组。

  • triples_factory (CoreTriplesFactory | None) – 三元组工厂。如果三元组是基于标签的，则必须提供。如果提供了且三元组是基于ID的，则向结果中添加标签。

  • batch_size (int | None) – 使用的批量大小。使用 None 以使用最大可能的批量大小。

  • mode (Literal['training', 'validation', 'testing'] | None) – 传递模式，在传导设置中为None，在归纳设置中为“training”、“validation”或“testing”之一。

Returns:

一个包含三元组及其预测得分的得分包。

Return type:

ScorePack