实例

class Instances[source]

基础类: Dataset[BatchType], Generic[SampleType, BatchType], ABC

训练实例的基类。

方法总结

from_triples(mapped_triples, *, ...)

从映射的三元组创建实例。

get_collator()

获取一个排序器。

方法文档

classmethod from_triples(mapped_triples: Tensor, *, num_entities: int, num_relations: int, **kwargs) → Instances[来源]

从映射的三元组创建实例。

Parameters:

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

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

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

  • kwargs – 额外的基于关键字的参数。

Returns:

实例。

Return type:

Instances

# noqa:DAR202 # noqa:DAR401

get_collator() → Callable[[list[SampleType]], BatchType] | None[来源]

获取一个排序器。

Return type:

Callable[[list[SampleType]], BatchType] | 无