fixed style error in NaiveBayes.scala

rekhajoshm · Jan 21, 2015 · 0313c0c · 0313c0c
1 parent 4a3676d
commit 0313c0c
Showing 1 changed file with 1 addition and 2 deletions.
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
@@ -59,8 +59,7 @@ class NaiveBayesModel private[mllib] (
   private val brzNegTheta: Option[BDM[Double]] = model match {
     case NaiveBayesModels.Multinomial => None
     case NaiveBayesModels.Bernoulli =>
-      val negTheta = brzLog((brzExp(brzTheta.copy) :*= (-1.0)) :+= 1.0)
-      //((x) => math.log(1.0 - math.exp(x))
+      val negTheta = brzLog((brzExp(brzTheta.copy) :*= (-1.0)) :+= 1.0) // log(1.0 - exp(x))
       Option(negTheta)
   }