EntityRelationEmbeddingModel¶

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.

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