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 entity representations.
The relation embedding dimension.
The relation representations.
Methods Summary
Has to be called after each parameter update.
Attributes Documentation
- entity_representations
The entity representations.
This property provides forward compatibility with the new-style
pykeen.models.ERModel
.- Return type
- relation_representations
The relation representations.
This property provides forward compatibility with the new-style
pykeen.models.ERModel
.- Return type
Methods Documentation