DistMA
- class DistMA(embedding_dim=256, entity_initializer=None, entity_initializer_kwargs=None, entity_normalizer=None, entity_normalizer_kwargs=None, relation_initializer=None, relation_initializer_kwargs=None, **kwargs)[source]
Bases:
ERModel
An implementation of DistMA from [shi2019].
Initialize DistMA via the
pykeen.nn.modules.DistMAInteraction
interaction.- Parameters:
embedding_dim (
int
) – The entity embedding dimension \(d\).entity_initializer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Entity initializer function. Defaults to Noneentity_initializer_kwargs (
Optional
[Mapping
[str
,Any
]]) – Keyword arguments to be used when calling the entity initializerentity_normalizer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Entity normalizer function. Defaults to Noneentity_normalizer_kwargs (
Optional
[Mapping
[str
,Any
]]) – Keyword arguments to be used when calling the entity normalizerrelation_initializer (
Union
[str
,Callable
[[FloatTensor
],FloatTensor
],None
]) – Relation initializer function. Defaults to Nonerelation_initializer_kwargs (
Optional
[Mapping
[str
,Any
]]) – Keyword arguments to be used when calling the relation initializerkwargs – Remaining keyword arguments passed through to
pykeen.models.ERModel
.
Attributes Summary
The default strategy for optimizing the model's hyper-parameters
Attributes Documentation