Why do I only get 3 classes in the inference results of AWS Comprehend Custom Classification

0

I trained a custom Multi-label classifier and created an endpoint for it. There are more than 20 classes all together. However, invoking the classifier I only ever get 3 classes / labels and their respective probabilities back in the result.

  response = comprehend.classify_document(Text=text, EndpointArn=arn)

response['Labels'] is always of length 3 (i assume the 3 best scoring labels).

Is there a way to get back all the labels and their probabilities ?

jan
質問済み 3ヶ月前178ビュー
2回答
1

Hello !

Please note that when we create a custom classifier for multi class, the custom comprehend model will detect only 3 top classes along with the scores.

Unfortunately, Comprehend does not have any feature as of now to allow the users to select the number of classes that we want to be detected for each file.

AWS
サポートエンジニア
回答済み 3ヶ月前
  • That's unfortunate, thanks for answering.

0

Actually i just got back a result with 4 classes (out of 32 possible classes) for the first time .. so it's not the top 3 classes after all .. How is the result determined then?

jan
回答済み 3ヶ月前

ログインしていません。 ログイン 回答を投稿する。

優れた回答とは、質問に明確に答え、建設的なフィードバックを提供し、質問者の専門分野におけるスキルの向上を促すものです。

質問に答えるためのガイドライン

関連するコンテンツ