Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
Showing results for 
Search instead for 
Did you mean: 

How to get the Probability of a prediction from Real Time Inference model

New Contributor II


I have used AutoML to create a model. When using that model I want to have the probability of the predictions returned. I was able to do this in a notebook with:

loaded_model = mlflow.pyfunc.load_model(logged_model)

# Predict on a Pandas DataFrame.

import pandas as pd


I have since registered the model and deployed a Real Time Inference endpoint for this model.

When calling the API I only get a true/false response like so:


"predictions": [




It is not clear to me how to change or update the model to return a probability. Any help would be much appreciated.


Join 100K+ Data Experts: Register Now & Grow with Us!

Excited to expand your horizons with us? Click here to Register and begin your journey to success!

Already a member? Login and join your local regional user group! If there isn’t one near you, fill out this form and we’ll create one for you to join!