langml.prompt.clf.utils

Module Contents

Classes

MetricsCallback

Functions

merge_template_tokens(template_ids: List[int], token_ids: List[int], max_length: Optional[int] = None) → Tuple[List[int], List[int]]

Merge template and token ids
langml.prompt.clf.utils.merge_template_tokens(template_ids: List[int], token_ids: List[int], max_length: Optional[int] = None) → Tuple[List[int], List[int]][source]

Merge template and token ids :param - template_ids: List[int], template ids :param - token_ids: List[int], token ids :param - max_length: int, max length

Returns

List[int], merged token ids - template_mask: List[int], template mask

Return type

  • token_ids

class langml.prompt.clf.utils.MetricsCallback(data: List[str], labels: List[str], mask_id: int, template: langml.prompt.base.Template, patience: int = 10, batch_size: int = 32, model_path: Optional[str] = None, f1_average: str = 'macro')[source]

Bases: langml.keras.callbacks.Callback

on_train_begin(self, logs=None)[source]
on_epoch_end(self, epoch, logs=None)[source]
on_train_end(self, logs=None)[source]