langml.prompt.base

Module Contents

Classes

Template
BasePromptModel
BasePromptTask
BaseDataGenerator

class langml.prompt.base.Template(template: List[str], label_tokens_map: Dict[str, List[str]], tokenizer: langml.tokenizer.Tokenizer)

__len__(self) → int

encode_template(self, template: str) → List[int]

encode_label_tokens_map(self, label_tokens_map: Dict[str, List[str]]) → Dict[str, List[int]]

decode_label(self, idx: int, default='<UNK>') → str

class langml.prompt.base.BasePromptModel(plm_backbone: str, plm_config_path: str, plm_ckpt_path: str, template: Template, learning_rate: float = 1e-05, freeze_plm: bool = True)

abstract build_model(self) → langml.tensor_typing.Models

class langml.prompt.base.BasePromptTask(prompt_model: BasePromptModel, tokenizer: langml.tokenizer.Tokenizer)

abstract fit(self)

abstract predict(self)

class langml.prompt.base.BaseDataGenerator

abstract make_iter(self, random: bool = False)

abstract __len__(self)

__call__(self, random: bool = False)