Predict probability on AWS SageMaker AutoPilot endpoint

Question

I am testing SageMaker AutoPilot in order to verify how good it is for regular use.

Up until now, it seems relatively easy to use it, it trained a model with good results and it was easy to create the endpoint. I would like to get the predicted label and its probability, in order to check if the prediciton is good. However, I could only get the label and I did not find anything about retrieving the probability (predict_proba).

Is there any way to get the probability? Thank you!

Answer 1

you can now get select the inference response for classification models :)

More info at https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-container-output.html