Skip to content

Commit

Permalink
removed old inaccurate comment from test suite for mllib naive bayes
Browse files Browse the repository at this point in the history
  • Loading branch information
leahmcguire committed Feb 26, 2015
1 parent 76e5b0f commit d9477ed
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit d9477ed

Please sign in to comment.