- 10527 Views
- 10 replies
- 4 kudos
Resolved! How to get probability score for each prediction from mlflow
I trained my model and was able to get the batch prediction from that model as specified below. But I want to also get the probability scores for each prediction. Do you have any idea? Thank you!logged_model = path_to_model# Load model as a PyFuncMod...
- 10527 Views
- 10 replies
- 4 kudos
Latest Reply
- 4 kudos
Now you can log the model using this parameter:mlflow.sklearn.log_model( ..., # the usual params pyfunc_predict_fn="predict_proba" ) which will return probabilities for the first class apparently when using the model for inference (e.g. when...
- 4 kudos