Cleaner

class Cleaner[source]

Bases: object

A cleanup method for ensuring that all entities are contained in the triples of the first split part.

Methods Summary

__call__(triples_groups, random_state)

Cleanup a list of triples array with respect to the first array.

cleanup_pair(reference, other, random_state)

Clean up one set of triples with respect to a reference set.

Methods Documentation

__call__(triples_groups: Sequence[Tensor], random_state: None | int | Generator) Sequence[Tensor][source]

Cleanup a list of triples array with respect to the first array.

Parameters:
Return type:

Sequence[Tensor]

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

Clean up one set of triples with respect to a reference set.

Parameters:
  • reference (Tensor) – the reference set of triples, which shall contain triples for all entities

  • other (Tensor) – the other set of triples

  • random_state (None | int | Generator) – the random state to use, if any randomized operations take place

Returns:

a pair (reference, other), where some triples of other may have been moved into reference

Return type:

tuple[Tensor, Tensor]