LCWA实例

class LCWAInstances(*, pairs: ndarray, compressed: csr_matrix)[source]

基础:Instances[tuple[Tensor, Tensor], tuple[Tensor, Tensor]]

LCWA的三元组及其索引的映射。

初始化LCWA实例。

Parameters:
  • pairs (np.ndarray) – 唯一的配对

  • 压缩的 (scipy.sparse.csr_matrix) – CSR格式的压缩三元组

方法总结

from_triples(mapped_triples, *, ...[, target])

从三元组创建LCWA实例。

方法文档

classmethod from_triples(mapped_triples: Tensor, *, num_entities: int, num_relations: int, target: int | None = None, **kwargs) Instances[source]

从三元组创建LCWA实例。

Parameters:
  • mapped_triples (Tensor) – 形状: (num_triples, 3) 基于ID的三元组。

  • num_entities (int) – 实体的数量。

  • num_relations (int) – 关系的数量。

  • target (int | None) – 要预测的列

  • kwargs – 关键字参数(被丢弃)

Returns:

实例。

Return type:

Instances