Pre-Trained Models

Support for loading and managing pretrained models.

This module provides an API to download and have ready to use pre-trained ScoringBasedEmbeddingModel.

load_pretrained_model(dataset, scoring_type)

Function to load a pretrained model.

Currently the available models are trained on “FB15K-237”, “WN18RR”, “YAGO310”, “FB15K” and “WN18” and have as scoring function “TransE”, “DistMult”, “ComplEx”, “HolE” and “RotatE”.