fixed typo in NaiveBayes

leahmcguire · leahmcguire · commit 2d0c1ba63184 · 2015-03-05T13:42:42.000-08:00
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] = {

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@ class NaiveBayesModel private[mllib] (`
`62`	`62`	`case NaiveBayes.Multinomial => (None, None)`
`63`	`63`	`case NaiveBayes.Bernoulli =>`
`64`	`64`	`val negTheta = brzLog((brzExp(brzTheta.copy) :*= (-1.0)) :+= 1.0) // log(1.0 - exp(x))`
`65`		`- (Option(negTheta), Option(brzSum(brzNegTheta, Axis._1)))`
	`65`	`+ (Option(negTheta), Option(brzSum(negTheta, Axis._1)))`
`66`	`66`	`}`
`67`	`67`
`68`	`68`	`override def predict(testData: RDD[Vector]): RDD[Double] = {`