Hi, I am trying to change the following code so as to get the probabilities of the input Vector on each class (instead of the class itself with the highest probability). I know that this is already available as part of the most recent release of Spark but I have to use Spark 1.1.0.
Any help is appreciated. override def predict(testData: Vector): Double = { > labels(brzArgmax(brzPi + brzTheta * testData.toBreeze)) > } https://github.com/apache/spark/blob/v1.1.0/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala *// Adamantios*