architxt.nlp.entity_extractor

architxt.nlp.entity_extractor#

Classes

EntityExtractor()

FlairEntityExtractor([model_name])

SpacyEntityExtractor([model_name])

class architxt.nlp.entity_extractor.EntityExtractor[source]#

Bases: ABC

async batch(sentences)[source]#
Return type:

AsyncIterator[AnnotatedSentence]

async enrich(sentences)[source]#
Return type:

AsyncIterator[AnnotatedSentence]

property name#
Return type:

str

class architxt.nlp.entity_extractor.FlairEntityExtractor(model_name='ner')[source]#

Bases: EntityExtractor

async batch(sentences, *, batch_size=128)[source]#
Return type:

AsyncIterator[AnnotatedSentence]

class architxt.nlp.entity_extractor.SpacyEntityExtractor(model_name='en_core_web_sm')[source]#

Bases: EntityExtractor

async batch(sentences, *, batch_size=128)[source]#
Return type:

AsyncIterator[AnnotatedSentence]

Contents