EntityEmbeddingModel¶

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

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.