notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

how to get "return_all_score"

Open imji0319 opened this issue 2 years ago • 0 comments
trafficstars

Hi, thank you for your support.

I made model using "AutoModelForSequenceClassification" of transformers and then deploy the model using sagemaker.huggingface. That is really amazing but have a little issue for me. I'd like to get all score for all label when I predict for new data. I found an option, return_all_score, for it in transformers. I'd like to use the option in sagemaker.huggingface.

스크린샷 2023-01-18 오전 10 29 59

you said "using transformers pipelines so I wonder how to use the option in your works.

imji0319 avatar Jan 18 '23 01:01 imji0319