CoverageSplitter

class CoverageSplitter[source]

基础类:Splitter

这个分割器贪婪地选择训练三元组,以确保每个实体都被覆盖,然后分割其余部分。

方法总结

split_absolute_size(mapped_triples, sizes, ...)

将三元组分割成干净的组。

方法文档

split_absolute_size(mapped_triples: Tensor, sizes: Sequence[int], random_state: Generator) Sequence[Tensor][source]

将三元组分割成干净的组。

此方法对三元组进行分区,即每个三元组恰好在一个组中。此外,它确保第一个组至少包含所有实体一次。

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

  • sizes (Sequence[int]) – 每个分割部分的绝对三元组数量。

  • random_state (Generator) – 用于分割的随机状态

Returns:

每个分割部分的基于ID的三元组序列。绝对值可能不同以确保约束。

Return type:

Sequence[Tensor]