CollapseGraphPairCombinator

class CollapseGraphPairCombinator[source]

Bases: GraphPairCombinator

This combinator merges all matching entity pairs into a single ID.

Methods Summary

process(mapped_triples, alignment, offsets)

Process the combined mapped triples.

Methods Documentation

process(mapped_triples, alignment, offsets)[source]

Process the combined mapped triples.

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

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

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

Return type:

ProcessedTuple

Returns:

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