MultimodalModel¶
- class MultimodalModel(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.EntityRelationEmbeddingModel
A multimodal KGE model.
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.Model
See also
Constructor of the base class
pykeen.models.EntityEmbeddingModel