ComplExLiteral

class ComplExLiteral(triples_factory, embedding_dim=50, input_dropout=0.2, loss=None, preferred_device=None, random_seed=None)[source]

Bases: pykeen.models.unimodal.complex.ComplEx, pykeen.models.base.MultimodalModel

An implementation of ComplexLiteral from [agustinus2018] based on the LCWA training approach.

Initialize the 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

Methods Summary

forward(h_indices, r_indices, t_indices)

Unified score function.

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]] = {}

The default parameters for the default loss function class

Methods Documentation

forward(h_indices, r_indices, t_indices)[source]

Unified score function.

Return type

FloatTensor