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...
There seems to be some reluctance to implement this:
https://github.com/delta-io/delta/issues/348But there is an open PR:
https://github.com/delta-io/delta/pull/371