EntityRelationEmbeddingModel¶

class EntityRelationEmbeddingModel(triples_factory, embedding_dim=50, relation_dim=None, loss=None, predict_with_sigmoid=False, preferred_device=None, random_seed=None, regularizer=None, entity_initializer=None, entity_initializer_kwargs=None, entity_normalizer=None, entity_normalizer_kwargs=None, entity_constrainer=None, entity_constrainer_kwargs=None, relation_initializer=None, relation_initializer_kwargs=None, relation_normalizer=None, relation_normalizer_kwargs=None, relation_constrainer=None, relation_constrainer_kwargs=None)[source]¶

Bases: pykeen.models.base.Model

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.

See also

Constructor of the base class pykeen.models.EntityEmbeddingModel

Attributes Summary

`embedding_dim`	The entity embedding dimension.
`relation_dim`	The relation embedding dimension.

Methods Summary

post_parameter_update()

Has to be called after each parameter update.

Attributes Documentation

embedding_dim¶

The entity embedding dimension.

Return type: int

relation_dim¶: The relation embedding dimension.

Methods Documentation

post_parameter_update()[source]¶

Has to be called after each parameter update.

Return type: None