fixed typo in NaiveBayes

rekhajoshm · Mar 5, 2015 · 2d0c1ba · 2d0c1ba
1 parent c298e78
commit 2d0c1ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala
@@ -62,7 +62,7 @@ class NaiveBayesModel private[mllib] (
     case NaiveBayes.Multinomial => (None, None)
     case NaiveBayes.Bernoulli =>
       val negTheta = brzLog((brzExp(brzTheta.copy) :*= (-1.0)) :+= 1.0) // log(1.0 - exp(x))
-      (Option(negTheta), Option(brzSum(brzNegTheta, Axis._1)))
+      (Option(negTheta), Option(brzSum(negTheta, Axis._1)))
   }
 
   override def predict(testData: RDD[Vector]): RDD[Double] = {