我訓練模型,能夠得到批預測模型如下指定。但我想也為每個預測得分的概率。你有什麼主意嗎?
謝謝你!
logged_model = path_to_model
# PyFuncModel負載模型。
loaded_model = mlflow.pyfunc.load_model (logged_model)
#熊貓DataFrame預測。
熊貓作為pd導入
loaded_model.predict (pd.DataFrame(數據)
嗨@Kaniz開羅
錯誤的說‘PyFuncModel’對象沒有屬性“predict_proba”。
正如上麵顯示的,我用以下加載模型
loaded_model = mlflow.pyfunc.load_model (logged_model)和錯誤。
經曆mlflow文檔後,我改變了它
loaded_model = mlflow.sklearn.load_model (logged_model)和它工作正常。
現在一切都好。謝謝你的時間。
賽義德
嗨@Kaniz開羅
對不起,劫持。
我的問題是,如果我從mlflow讀取注冊模型,我隻能看見.predict方法但不是.predict_proba的選項。
我們有簡單的解決方案得到的概率?
謝謝
賽義德