apache-spark - Spark - 预测概率 API (predict_proba)

Question

Apache Spark 是否提供类似于 scikit-learn 的 predict_proba 函数(即 decision_tree.predict_proba(X))的 API 来获取决策树的预测概率？

Answer 1

我自己也在找这个。当我注意到 api 以一种非常尴尬的方式(至少对于 LogisticRegressionModel)具有功能时，我差点想出一个解决方案:

您清除阈值(使用 clearThreshold())。这样预测函数就不会返回标签，而是返回基础值。

Java docs这么说:

public LogisticRegressionModel clearThreshold() :: Experimental :: Clears the threshold so that predict will output raw prediction scores.

仅供引用:返回值介于 0. 和 1. 之间，默认阈值为 .5，因此您可以轻松评估要将阈值设置为什么。