dicee.models.ensemble

Classes

Module Contents

class dicee.models.ensemble.EnsembleKGE(models: list = None, seed_model=None, pretrained_models: List = None)

name

train_mode = True

args

named_children()

property example_input_array

parameters()

modules()

__iter__()

__len__()

eval()

to(device)

state_dict(): Return the state dict of the ensemble.

load_state_dict(state_dict, strict=True): Load the state dict into the ensemble.

mem_of_model()

__call__(x_batch)

step()

get_embeddings()

__str__()