Prediction Probability from mleap model deployed on sagemaker

[janvekarnaveed]

After deploying mleap model artifact on sagemaker endpoint, during the inferencing stage, how can I get probabilities of prediction rather than binary outcome? I tried changing output column from prediction to probability, but I get the following error:
sagemaker: ml.combust.mleap.tensor.DenseTensor cannot be cast to java.lang.Double

Any suggestions?

[ancasarb]

Looks like this is answered under aws/sagemaker-sparkml-serving-container#13 (comment), closing for now, please re-open if you have further questions.