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

jat*_*eet 5 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中直接从外部访问这些变量.

谢谢,贾丁