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

Question

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 ?

Answer

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.

Answer

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?

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

Relevant content