nlp_architect.api package
Submodules
nlp_architect.api.abstract_api module
nlp_architect.api.base module
-
class
nlp_architect.api.base.ModelAPI(model_path: str = None)[source] Bases:
objectBase class for a model API implementation Implementing classes must provide a default model and/or a path to a model
Parameters: model_path (str) – path to a trained model run method must return
-
default_model= None
-
nlp_architect.api.bist_parser_api module
-
class
nlp_architect.api.bist_parser_api.BistParserApi[source] Bases:
nlp_architect.api.abstract_api.AbstractApiBist Parser API
-
inference(doc)[source] Parse according to SpacyBISTParser’s model
Parameters: doc (str) – the doc str Returns: the parser’s response hosted in CoreNLPDoc object Return type: CoreNLPDoc
-
nlp_architect.api.intent_extraction_api module
-
class
nlp_architect.api.intent_extraction_api.IntentExtractionApi(prompt=False)[source] Bases:
nlp_architect.api.abstract_api.AbstractApi-
model_dir= '/home/runner/nlp-architect/cache/intent-pretrained'
-
pretrained_model= '/home/runner/nlp-architect/cache/intent-pretrained/model.h5'
-
pretrained_model_info= '/home/runner/nlp-architect/cache/intent-pretrained/model_info.dat'
-
nlp_architect.api.ner_api module
-
class
nlp_architect.api.ner_api.NerApi(prompt=False)[source] Bases:
nlp_architect.api.abstract_api.AbstractApiNER model API
-
model_dir= '/home/runner/nlp-architect/cache/ner-pretrained'
-
pretrained_model= '/home/runner/nlp-architect/cache/ner-pretrained/model_v4.h5'
-
pretrained_model_info= '/home/runner/nlp-architect/cache/ner-pretrained/model_info_v4.dat'
-