EntityEmbeddingModel¶
- class EntityEmbeddingModel(*, triples_factory, entity_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 most KGE models that have one embedding for entities.
Initialize the entity embedding model.
See also
Constructor of the base class
pykeen.models.Model
Attributes Summary
The entity embedding dimension.
The entity 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
Methods Documentation