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].

Attributes Summary

`head_entities`
`tail_entities`
`testing_set`
`training_set`

Methods Summary

`eval_method`(data)
`get_embeddings`()
`init_module`()
`load_best_model`()
`load_eval_data`()
`train`()	Abstract method for training the model.

Attributes Documentation

head_entities

tail_entities

testing_set

training_set

Methods Documentation

eval_method(data)[source]

get_embeddings()[source]

init_module()[source]

load_best_model()[source]

load_eval_data()[source]

train()[source]: Abstract method for training the model. This method must be implemented in children classes