SwapGraphPairCombinator

class SwapGraphPairCombinator

Bases: GraphPairCombinator[FactoryType]

Add extra triples by swapping aligned entities.

Methods Summary

process(mapped_triples, alignment, offsets)

Process the combined mapped triples.

Methods Documentation

process(mapped_triples: Tensor, alignment: Tensor, offsets: Tensor) → ProcessedTuple

Process the combined mapped triples.

Parameters:

• mapped_triples (Tensor) – shape: (n, 3) the id-based merged triples

• alignment (Tensor) – shape: (2, m) the id-based entity alignment

• offsets (Tensor) – shape: (2, 2) the entity and relation offsets from merging

Returns:

updated triples and alignment tensor, as well as parameters for updating label-to-id mappings

Return type:

ProcessedTuple