ELBoxEmbeddings

class mowl.models.ELBoxEmbeddings(dataset, embed_dim=50, margin=0, reg_norm=1, learning_rate=0.001, epochs=1000, batch_size=32768, model_filepath=None, device='cpu')[source]

Bases: EmbeddingELModel

Implementation based on [peng2020].

Methods Summary

eval_method(data)

get_embeddings()

init_module()

load_best_model()

train()

Abstract method for training the model.

Methods Documentation

eval_method(data)[source]
get_embeddings()[source]
init_module()[source]
load_best_model()[source]
train()[source]

Abstract method for training the model. This method must be implemented in children classes