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.CrossEInteractioninteraction.- 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