Spark - Prediction Probability API（predict_proba）

Question

Apache Spark是否提供API來獲取類似於scikit-learn的predict_proba函數（即decision_tree.predict_proba（X））的決策樹預測概率？

Answer 1

早上好，

此鏈接可以幫助你：http://spark.apache.org/docs/latest/mllib-decision-tree.html

Answer 2

我自己尋找這一點。當我注意到api具有非常尷尬的功能時（至少對於LogisticRegressionModel），我差點破解了一個解決方案：

您清除了閾值（使用clearThreshold()）。這樣預測功能不會返回標籤，而是返回底層的價值。

Java docs這樣說：

公共LogisticRegressionModel clearThreshold（） ::實驗::清除閾值，使得預測將要輸出的原始預測分數。

僅供參考：返回值介於0和1之間，默認閾值爲0.5，因此您可以輕鬆評估要設置閾值的內容。