如何从Spark ML OneVsRest分类器获取原始概率?

时间:2016-11-15 14:00:43

标签: java apache-spark apache-spark-mllib apache-spark-ml

我想使用one vs rest(all)解决多类分类问题。我在Spark ML中找到了OneVsRest,它将多类问题分解为几个二进制分类。我希望有一种获得原始概率的方法,但我似乎无法找到它。在mllib中,我可以使用clearThreshold()来创建二元分类器,如逻辑回归返回原始概率,但我没有看到ML逻辑回归中可用的方法。我确实看到rawPredictionCol()返回一个参数,但没有关于如何使用它的文档。我正在使用Java spark API,并尝试将此example用于产品原始概率。问题2:由于某种原因,我总是发现Spark文档缺少太多的案例..我错过了什么?

更新:尝试阅读源代码..这不是一个非常有效的练习;我不是Scala程序员。将根据调查结果进行更新。

0 个答案:

没有答案