ontolearn.nces_trainer

Trainer for NCES instances

Classes

NCESTrainer

Trainer for neural class expression synthesizers, i.e., NCES, NCES2, ROCES.

Functions

before_pad(arg)

Module Contents

ontolearn.nces_trainer.before_pad(arg)

class ontolearn.nces_trainer.NCESTrainer(synthesizer, epochs=300, batch_size=128, learning_rate=0.0001, tmax=20, eta_min=1e-05, clip_value=5.0, num_workers=8, storage_path='./')

Trainer for neural class expression synthesizers, i.e., NCES, NCES2, ROCES.

synthesizer

epochs = 300

batch_size = 128

learning_rate = 0.0001

tmax = 20

eta_min = 1e-05

clip_value = 5.0

num_workers = 8

storage_path = './'

static compute_accuracy(prediction, target)

get_optimizer(model, emb_model=None, optimizer='Adam')

get_data_idxs()

get_er_vocab()

static show_num_trainable_params(synthesizer)

collate_batch(batch)

map_to_token(idx_array)

train_step(batch, model, emb_model, optimizer, device, triples_dataloader=None)

train(data, shuffle_examples=False, example_sizes=None, save_model=True, optimizer='Adam', record_runtime=True)