EntityRelationEmbeddingModel¶
-
class
EntityRelationEmbeddingModel
(triples_factory, entity_representations, relation_representations, loss=None, predict_with_sigmoid=False, preferred_device=None, random_seed=None, regularizer=None)[source]¶ Bases:
pykeen.models.base._OldAbstractModel
,abc.ABC
A base module for KGE models that have different embeddings for entities and relations.
Initialize the entity embedding model.
See also
Constructor of the base class
pykeen.models.Model
Attributes Summary
The entity embedding dimension.
The relation embedding dimension.
Methods Summary
Has to be called after each parameter update.
Attributes Documentation
-
relation_dim
¶ The relation embedding dimension.
Methods Documentation
-