Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

S

spark
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents

Snippets Groups Projects

f00e949f

Commit f00e949f authored 11 years ago by Matei Zaharia

Downloads
- Patches
- Plain Diff

Added Java unit test, data, and main method for Naive Bayes

Also fixes mains of a few other algorithms to print the final model

parent 4c28a2ba

No related branches found

No related tags found

No related merge requests found

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

mllib/data/sample_naive_bayes_data.txt 6 additions, 0 deletions

mllib/data/sample_naive_bayes_data.txt
mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala 3 additions, 1 deletion

...pache/spark/mllib/classification/LogisticRegression.scala
mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala 20 additions, 1 deletion

...la/org/apache/spark/mllib/classification/NaiveBayes.scala
mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala 2 additions, 0 deletions

...ain/scala/org/apache/spark/mllib/classification/SVM.scala
mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala 3 additions, 1 deletion

.../main/scala/org/apache/spark/mllib/regression/Lasso.scala
mllib/src/main/scala/org/apache/spark/mllib/regression/LinearRegression.scala 2 additions, 0 deletions

.../org/apache/spark/mllib/regression/LinearRegression.scala
mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala 3 additions, 1 deletion

...a/org/apache/spark/mllib/regression/RidgeRegression.scala
mllib/src/test/java/org/apache/spark/mllib/classification/JavaNaiveBayesSuite.java 72 additions, 0 deletions

...pache/spark/mllib/classification/JavaNaiveBayesSuite.java

with 111 additions and 4 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment