dicee.models.ensemble

Classes

Module Contents

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

name

train_mode = True

args

named_children()[source]

property example_input_array

parameters()[source]

modules()[source]

__iter__()[source]

__len__()[source]

eval()[source]

to(device)[source]

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

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

mem_of_model()[source]

__call__(x_batch)[source]

step()[source]

get_embeddings()[source]

__str__()[source]