SLCWA实例

class SLCWAInstances(*, mapped_triples: Tensor, num_entities: int | None = None, num_relations: int | None = None, negative_sampler: str | NegativeSampler | type[NegativeSampler] | None = None, negative_sampler_kwargs: Mapping[str, Any] | None = None)

基础类：Instances[tuple[Tensor, Tensor, Tensor | None], SLCWABatch]

sLCWA的训练实例。

初始化sLCWA实例。

Parameters:

• mapped_triples (MappedTriples) – 形状: (num_triples, 3) 基于ID的三元组，传递给负采样器

• num_entities (int | None) – >0 实体数量，传递给负采样器

• num_relations (int | None) – >0 关系的数量，传递给负采样器

• negative_sampler (HintOrType[NegativeSampler]) – 负采样器，或其提示

• negative_sampler_kwargs (OptionalKwargs) – 传递给负采样器的额外基于关键字的参数

方法总结

collate(samples)	整理样本。
from_triples(mapped_triples, *, ...)	从映射的三元组创建实例。
get_collator()	获取一个排序器。

方法文档

static collate(samples: Iterable[tuple[Tensor, Tensor, Tensor | None]]) → SLCWABatch

整理样本。

Parameters:

样本 (可迭代[元组[张量, 张量, 张量 | 无]])

Return type:

SLCWABatch

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[tuple[Tensor, Tensor, Tensor | None]]], SLCWABatch] | None

获取一个排序器。

Return type:

Callable[[list[tuple[Tensor, Tensor, Tensor | None]]], SLCWABatch] | None