MuRE¶

class MuRE(*, embedding_dim=200, p=2, power_norm=True, entity_initializer=<function normal_>, entity_initializer_kwargs=None, entity_bias_initializer=<function zeros_>, relation_initializer=<function normal_>, relation_initializer_kwargs=None, relation_matrix_initializer=<function uniform_>, relation_matrix_initializer_kwargs=None, **kwargs)[source]¶

Bases: Generic[pykeen.typing.HeadRepresentation, pykeen.typing.RelationRepresentation, pykeen.typing.TailRepresentation], pykeen.models.nbase._NewAbstractModel

An implementation of MuRE from [balazevic2019b].

Initialize MuRE via the pykeen.nn.modules.MuREInteraction interaction.

Parameters

embedding_dim (int) – The entity embedding dimension \(d\). Defaults to 200. Is usually \(d \in [50, 300]\).
p (int) – The \(l_p\) norm. Defaults to 2.
power_norm (bool) – Should the power norm be used? Defaults to true.

Attributes Summary

hpo_default

The default strategy for optimizing the model’s hyper-parameters

Attributes Documentation

hpo_default: ClassVar[Mapping[str, Any]] = {'embedding_dim': {'high': 256, 'low': 16, 'q': 16, 'type': <class 'int'>}, 'p': {'high': 2, 'low': 1, 'type': <class 'int'>}}¶: The default strategy for optimizing the model’s hyper-parameters