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

Bases: pykeen.models.base.EntityEmbeddingModel

A base module for KGE models that have different embeddings for entities and relations.

Initialize the entity embedding model.


relation_dim (Optional[int]) – The relation embedding dimensionality. If not given, defaults to same size as entity embedding dimension.

See also

Constructor of the base class pykeen.models.Model

See also

Constructor of the base class pykeen.models.EntityEmbeddingModel