来自Spark Gradient Boosted Trees的预测概率(类概率)(版本1.6.1 / 1.5.2)

时间:2016-04-27 06:48:33

标签: apache-spark prediction gbm

Is it possible to obtain class probabilities using GradientBoostedTrees with spark mllib?

获取提示

这个问题的答案是去年12月。我想知道Spark 1.6.1中是否有任何内容。

我试图编辑上面的问题,但不知怎的,它没有发生。

我的查询与Spark mllib库中的大多数其他算法一样,我们可以找到预测的概率。我们可以在Gradient Boosting Trees-Classification(http://spark.apache.org/docs/latest/mllib-ensembles.html#gradient-boosted-trees-vs-random-forests)中获得相同的结果并更改模型阈值吗?

这样我就可以改变阈值来改善模型性能。

0 个答案:

没有答案