EntityEmbeddingModel

class EntityEmbeddingModel(triples_factory, embedding_dim=50, 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)

Bases: pykeen.models.base.Model

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

Initialize the entity embedding model.

Parameters

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

See also

Constructor of the base class pykeen.models.Model

Attributes Summary

embedding_dim

The entity 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

Methods Documentation

post_parameter_update()

Has to be called after each parameter update.

Return type

None