python - 有没有办法在 AWS Comprehend 中使用两个经过训练的模型来检测实体？

Question

我有两个训练有素的实体识别器模型，它们检测不同的实体类型，我为这两个模型创建了端点，但我似乎找不到可以实现这两个模型以从简单的字符串文本中检测实体的用法。

一种模型从文本中检测客户（如“耐克”）和其他指标（如“销售额”）[如下所示]。

text="find net sales of Nike"
comprehend.detect_entities(Text=text,LanguageCode="en",EndpointArn='arn:aws:comprehend:xyz-1:760825412989:entity-recognizer-endpoint/xyz-1')

有没有办法在给定文本上使用端点来实现这两个模型？

Answer 1

Comprehend 支持每个请求针对一个模型进行推理。如果您有多个模型，您可以在 lambda 中链接推理请求并按顺序调用模型。

使用上面的示例代码：

text="find net sales of Nike"
comprehend.detect_entities(Text=text,LanguageCode="en",EndpointArn='arn:aws:comprehend:region:760825412989:entity-recognizer-endpoint/xyz-1')
comprehend.detect_entities(Text=text,LanguageCode="en",EndpointArn='arn:aws:comprehend:region:760825412989:entity-recognizer-endpoint/xyz-2')