dicee.static_funcs_training

Functions

make_iterable_verbose(→ Iterable)

evaluate_lp([model, triple_idx, num_entities, ...])

evaluate_bpe_lp(model, triple_idx, ...[, info])

efficient_zero_grad(model)

Module Contents

dicee.static_funcs_training.make_iterable_verbose(iterable_object, verbose, desc='Default', position=None, leave=True) → Iterable[source]
dicee.static_funcs_training.evaluate_lp(model=None, triple_idx=None, num_entities=None, er_vocab: Dict[Tuple, List] = None, re_vocab: Dict[Tuple, List] = None, info='Eval Starts', batch_size=128, chunk_size=1000)[source]
dicee.static_funcs_training.evaluate_bpe_lp(model, triple_idx: List[Tuple], all_bpe_shaped_entities, er_vocab: Dict[Tuple, List], re_vocab: Dict[Tuple, List], info='Eval Starts')[source]
dicee.static_funcs_training.efficient_zero_grad(model)[source]