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
embedding_dim (
int
) – The entity embedding dimension \(d\). Defaults to 50.combination_activation (
Union
[str
,Module
,Type
[Module
],None
]) – The combination activation function.combination_activation_kwargs (
Optional
[Mapping
[str
,Any
]]) – Additional keyword-based arguments passed to the constructor of the combination activation function (if not already instantiated).combination_dropout (
Optional
[float
]) – An optional dropout applied to the combination.entity_initializer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Entity initializer function. Defaults topykeen.nn.init.xavier_uniform_()
relation_initializer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Relation initializer function. Defaults topykeen.nn.init.xavier_uniform_()
relation_interaction_initializer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Relation interaction vector initializer function. Defaults topykeen.nn.init.xavier_uniform_()
kwargs – Remaining keyword arguments passed through to
pykeen.models.ERModel
.
Attributes Summary
The default strategy for optimizing the model's hyper-parameters
Attributes Documentation