如何從mlflow概率得分為每個預測

[Zoumana]

我訓練模型,能夠得到批預測模型如下指定。但我想也為每個預測得分的概率。你有什麼主意嗎?

謝謝你！

logged_model = path_to_model

# PyFuncModel負載模型。

loaded_model = mlflow.pyfunc.load_model (logged_model)

#熊貓DataFrame預測。

熊貓作為pd導入

loaded_model.predict (pd.DataFrame(數據)

[OndrejHavlicek]

現在您可以使用這個參數對數模型:

mlflow.sklearn。log_model (…,# the usual params pyfunc_predict_fn="predict_proba" )

將返回第一階級顯然當使用概率推理的模型(例如,當裝載它使用mlflow.pyfunc.spark_udf ())。