Github user sabhyankar commented on a diff in the pull request:
https://github.com/apache/spark/pull/8241#discussion_r37410650
--- Diff:
mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala ---
@@ -83,9 +86,13 @@ class NaiveBayesModel private[spark] (
}
override def predict(testData: RDD[Vector]): RDD[Double] = {
- val bcModel = testData.context.broadcast(this)
+ bcModel match {
--- End diff --
That makes sense. I initially started with just this Jira and once the PR
was ready, I realized the rest of them had similar fixes! Let me put something
together for this one and if it looks good, i will reproduce for the other PRs.
I will update this PR shortly. Thx!
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
