确定性清理器

class DeterministicCleaner[source]

基础类:Cleaner

清理一个三元组数组(测试)相对于另一个(训练)。

方法总结

cleanup_pair(reference, other, random_state)

清理一组三元组相对于参考集。

方法文档

cleanup_pair(reference: Tensor, other: Tensor, random_state: None | int | Generator) tuple[Tensor, Tensor][来源]

清理一组三元组相对于参考集。

Parameters:
  • 参考 (Tensor) – 三元组的参考集,应包含所有实体的三元组

  • 其他 (Tensor) – 另一组三元组

  • random_state (None | int | Generator) – 如果进行任何随机操作,则使用的随机状态

Returns:

一对(reference, other),其中other的一些三元组可能已被移动到reference中

Return type:

tuple[Tensor, Tensor]