ontolearn.clip_trainer

Module Contents

Classes

CLIPTrainer

CLIP trainer.

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

CLIP trainer.

compute_eval_metric(target, prediction)

get_optimizer(length_predictor, optimizer='Adam')

show_num_learnable_params()

train(train_dataloader, save_model=True, optimizer='Adam', record_runtime=True)