EntityRelationEmbeddingModel

class EntityRelationEmbeddingModel(*, triples_factory, entity_representations=None, entity_representations_kwargs=None, relation_representations=None, relation_representations_kwargs=None, **kwargs)[source]

Bases: pykeen.models.base._OldAbstractModel, abc.ABC

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

Initialize the entity embedding model.

See also

Constructor of the base class pykeen.models.Model

Attributes Summary

`embedding_dim`	The entity embedding dimension.
`entity_representations`	The entity representations.
`relation_dim`	The relation embedding dimension.
`relation_representations`	The relation representations.

Methods Summary

post_parameter_update()

Has to be called after each parameter update.

Attributes Documentation

embedding_dim

The entity embedding dimension.

Return type: int

entity_representations

The entity representations.

This property provides forward compatibility with the new-style pykeen.models.ERModel.

Return type: Sequence[Representation]

relation_dim

The relation embedding dimension.

Return type: int

relation_representations

The relation representations.

This property provides forward compatibility with the new-style pykeen.models.ERModel.

Return type: Sequence[Representation]

Methods Documentation

post_parameter_update()[source]

Has to be called after each parameter update.

Return type: None