class EntityEmbeddingModel(triples_factory, embedding_dim=50, loss=None, predict_with_sigmoid=False, automatic_memory_optimization=None, preferred_device=None, random_seed=None, regularizer=None)[source]

Bases: pykeen.models.base.Model

A base module for most KGE models that have one embedding for entities.

Initialize the entity embedding model.


embedding_dim (int) – The embedding dimensionality. Exact usages depends on the specific model subclass.

See also

Constructor of the base class pykeen.models.Model