CrossE

class CrossE(*, embedding_dim=50, combination_activation=<class 'torch.nn.modules.activation.Tanh'>, combination_activation_kwargs=None, combination_dropout=0.5, entity_initializer=<function xavier_uniform_>, relation_initializer=<function xavier_uniform_>, relation_interaction_initializer=<function xavier_uniform_>, **kwargs)[source]

Bases: ERModel[FloatTensor, Tuple[FloatTensor, FloatTensor], FloatTensor]

An implementation of CrossE from [zhang2019b].

Initialize CrossE via the pykeen.nn.modules.CrossEInteraction interaction.

Parameters:

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'>}}

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