I’ve deployed the Hate-speech-CNERG/bert-base-uncased-hatexplain multi-class text classification model (along with others) to a SageMaker endpoint. Using the TextClassificationPipeline you’re able to pass in the return_all_scores=True parameter to see all class labels for the input. However when us…

Return all class labels from SageMaker invoke_endpoint

lamwilton January 22, 2022, 5:01am 9

It finally worked. Thanks Marshmellow!

1 Like

Topic		Replies	Views
Return label_id, label and score from SageMaker Endpoint Amazon SageMaker	0	286	October 6, 2023
Emotion Model: Additional inference parameter not processed in Sagemaker Amazon SageMaker	4	1216	June 28, 2022
Emotion Model: Additional inference parameter not processed in Sagemaker inferentia instance Amazon SageMaker	1	279	July 17, 2023
Missing likelihood for all the classes in inference after BERT model deployed in SageMaker Beginners	1	530	June 15, 2022
Sagemaker Huggingface zero shot classification with candidate_labels in predict method Amazon SageMaker	2	1886	June 16, 2022