Skip to content

archaeo_super_prompt.modeling.entity_extractor

[docs] package archaeo_super_prompt.modeling.entity_extractor 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
"""Root of the module for infering in the NER model.

The purpose of this model is to extract hints about chunks for helping the
final LLM model to extract some named values for some fields.
"""

from .ner_transformer import NerModel
from .ne_selector import NeSelector
from .types import ChunksWithThesaurus, NamedEntityField

__all__ = ["NerModel", "NeSelector", "ChunksWithThesaurus", "NamedEntityField"]