DeterministicCleaner

class DeterministicCleaner

Bases: Cleaner

Cleanup a triples array (testing) with respect to another (training).

Methods Summary

cleanup_pair(reference, other, random_state)

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

Methods Documentation

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

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]