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.
Methods Summary
Has to be called after each parameter update.
Attributes Documentation
Methods Documentation