-
- Downloads
[SPARK-14077][ML] Refactor NaiveBayes to support weighted instances
## What changes were proposed in this pull request? 1,support weighted data 2,use dataset/dataframe instead of rdd 3,make mllib as a wrapper to call ml ## How was this patch tested? local manual tests in spark-shell unit tests Author: Zheng RuiFeng <ruifengz@foxmail.com> Closes #12819 from zhengruifeng/weighted_nb.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/NaiveBayes.scala 117 additions, 37 deletions...scala/org/apache/spark/ml/classification/NaiveBayes.scala
- mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala 26 additions, 73 deletions...la/org/apache/spark/mllib/classification/NaiveBayes.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/NaiveBayesSuite.scala 48 additions, 2 deletions.../org/apache/spark/ml/classification/NaiveBayesSuite.scala
Loading
Please register or sign in to comment