ELBE

class mowl.models.ELBE(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`([epochs, validate_every])	Abstract method for training the model.

Methods Documentation

eval_method(data)[source]

get_embeddings()[source]

init_module()[source]

load_best_model()[source]

train(epochs=None, validate_every=1)[source]: Abstract method for training the model. This method must be implemented in children classes