RelationLossWeighter

class RelationLossWeighter(weights: Tensor)

Bases: LossWeighter

Determine loss weights based solely on the relation.

Initialize the weighter.

Parameters:

weights (Tensor) – shape: (num_relations,) The weight per relation.

Methods Summary

__call__(h, r, t)	Calculate the sample weights for the given triples.
inverse_relation_frequency(mapped_triples)	Create a loss weighter with inverse relation frequencies.

Methods Documentation

__call__(h: Tensor | None, r: Tensor | None, t: Tensor | None) → Tensor

Calculate the sample weights for the given triples.

Does support broadcasting semantics.

Parameters:

• h (Tensor | None) – The head indices, or None to denote all of them.

• r (Tensor | None) – The relation indices, or None to denote all of them.

• t (Tensor | None) – The tail indices, or None to denote all of them.

Returns:

The sample weights.

Return type:

Tensor

classmethod inverse_relation_frequency(mapped_triples: Tensor) → Self

Create a loss weighter with inverse relation frequencies.

Parameters:

mapped_triples (Tensor)

Return type:

Self