ontolearn.clip_trainer

Classes

CLIPTrainer

CLIP trainer.

Module Contents

class ontolearn.clip_trainer.CLIPTrainer(clip, epochs=300, learning_rate=0.0001, decay_rate=0, clip_value=5.0, storage_path='./')[source]

CLIP trainer.

clip

epochs

learning_rate

decay_rate

clip_value

storage_path

compute_eval_metric(target, prediction)[source]

get_optimizer(length_predictor, optimizer='Adam')[source]

show_num_learnable_params()[source]

train(train_dataloader, save_model=True, optimizer='Adam', record_runtime=True)[source]