From d9477ed8450594de9f2da24af8f82c82def5ce24 Mon Sep 17 00:00:00 2001 From: leahmcguire Date: Thu, 26 Feb 2015 09:16:12 -0800 Subject: [PATCH] removed old inaccurate comment from test suite for mllib naive bayes --- .../org/apache/spark/mllib/classification/NaiveBayesSuite.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/test/scala/org/apache/spark/mllib/classification/NaiveBayesSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/classification/NaiveBayesSuite.scala index 853e4501d8685..29871215f9deb 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/classification/NaiveBayesSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/classification/NaiveBayesSuite.scala @@ -143,7 +143,7 @@ class NaiveBayesSuite extends FunSuite with MLlibTestSparkContext { val testRDD = sc.parallelize(testData, 2) testRDD.cache() - val model = NaiveBayes.train(testRDD, 1.0, "Bernoulli") ///!!! this gives same result on both models check the math + val model = NaiveBayes.train(testRDD, 1.0, "Bernoulli") validateModelFit(pi, theta, model) val validationData = NaiveBayesSuite.generateNaiveBayesInput(pi, theta, nPoints, 20, NaiveBayesModels.Bernoulli)