langml.baselines.ner

Submodules

• langml.baselines.ner.bert_crf
• langml.baselines.ner.cli
• langml.baselines.ner.dataloader
• langml.baselines.ner.lstm_crf

Package Contents

Classes

Infer

Functions

bio_decode(tags: List[str]) → List[Tuple[int, int, str]]	Decode BIO tags
compute_detail_metrics(model: langml.tensor_typing.Models, dataloader: object, id2label: Dict, is_bert: bool = True)

Attributes

TF_VERSION
Models
re_split

langml.baselines.ner.TF_VERSION

langml.baselines.ner.bio_decode(tags: List[str]) → List[Tuple[int, int, str]]

Decode BIO tags

Examples: >>> bio_decode([‘B-PER’, ‘I-PER’, ‘O’, ‘B-ORG’, ‘I-ORG’, ‘I-ORG’]) >>> [(0, 1, ‘PER’), (3, 5, ‘ORG’)]

langml.baselines.ner.Models

langml.baselines.ner.re_split

class langml.baselines.ner.Infer(model: langml.tensor_typing.Models, tokenizer: object, id2label: Dict, max_chunk_len: Optional[int] = None, is_bert: bool = True)

decode_one(self, text: str, base_position: int = 0)

Parameters

• text (-) – str

• base_position (-) – int

Returns

[(entity, start, end, entity_type)]

Return type

list of tuple

__call__(self, text: str)

langml.baselines.ner.compute_detail_metrics(model: langml.tensor_typing.Models, dataloader: object, id2label: Dict, is_bert: bool = True)