DistMultLiteral

class DistMultLiteral(triples_factory, embedding_dim=50, input_dropout=0.0, **kwargs)[source]

Bases: pykeen.models.multimodal.base.LiteralModel

An implementation of the LiteralE model with the DistMult interaction from [kristiadi2018].

Initialize the module.

Parameters

  • triples_factory (TriplesNumericLiteralsFactory) – The triples factory facilitates access to the dataset.

  • interaction – The interaction module (e.g., TransE)

  • interaction_kwargs – Additional key-word based parameters given to the interaction module’s constructor, if not already instantiated.

  • entity_representations – The entity representation or sequence of representations

  • entity_representations_kwargs – additional keyword-based parameters for instantiation of entity representations

  • relation_representations – The relation representation or sequence of representations

  • relation_representations_kwargs – additional keyword-based parameters for instantiation of relation representations

  • skip_checks – whether to skip entity representation checks.

  • kwargs – Keyword arguments to pass to the base model

Attributes Summary

hpo_default

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

loss_default_kwargs

The default parameters for the default loss function class

Attributes Documentation

hpo_default: ClassVar[Mapping[str, Any]] = {'embedding_dim': {'high': 256, 'low': 16, 'q': 16, 'type': <class 'int'>}, 'input_dropout': {'high': 0.5, 'low': 0.0, 'q': 0.1, 'type': <class 'float'>}}

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

loss_default_kwargs: ClassVar[Mapping[str, Any]] = {'margin': 0.0}

The default parameters for the default loss function class