在Spark的MLlib中获取Naive Baye的后验概率

时间:2014-11-26 06:48:53

标签: apache-spark apache-spark-mllib

我试图使用Java访问使用MLlib进行Naive Baye预测的后验概率。由于成员变量brzPi和brzTheta是私有的,我应用了一个hack来通过反射访问值。

我正在使用Java,但找不到使用Java的breeze库的方法。如果我是正确的,相关的计算是通过NaiveBayesModel类中的第66行给出的,

标签(brzArgmax(brzPi + brzTheta * testData.toBreeze))

这里DenseVectors的元素添加和乘法是作为不能在Java中直接访问的运算符给出的。另外,对我来说,使用brzArgmax并不十分清楚。

任何人都可以帮我将上述计算从Scala转换为Java。

PS:我已经在Jira上提出了一项改进请求,要求在MLlib中直接从外部访问这些变量。

谢谢, Jatin

0 个答案:

没有答案