Rename filename to match trait name, rearranged imports, added additi…

…onal comments
apache · Aug 26, 2015 · 965aaec · 965aaec
1 parent ab175b4
commit 965aaec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/NaiveBayes.scala
@@ -21,15 +21,16 @@ import java.lang.{Iterable => JIterable}
 
 import scala.collection.JavaConverters._
 
+import org.json4s.JsonDSL._
+import org.json4s.jackson.JsonMethods._
+
 import org.apache.spark.{Logging, SparkContext, SparkException}
 import org.apache.spark.annotation.Since
 import org.apache.spark.mllib.linalg.{BLAS, DenseMatrix, DenseVector, SparseVector, Vector}
 import org.apache.spark.mllib.regression.LabeledPoint
 import org.apache.spark.mllib.util.{Broadcastable, Loader, Saveable}
 import org.apache.spark.rdd.RDD
 import org.apache.spark.sql.{DataFrame, SQLContext}
-import org.json4s.JsonDSL._
-import org.json4s.jackson.JsonMethods._
 
 /**
  * Model for Naive Bayes Classifiers.

diff --git a/...che/spark/mllib/util/ModelBroadcast.scala → ...ache/spark/mllib/util/Broadcastable.scala b/...che/spark/mllib/util/ModelBroadcast.scala → ...ache/spark/mllib/util/Broadcastable.scala
@@ -29,6 +29,8 @@ private[spark] trait Broadcastable[T] {
   /**
    * Checks whether the model object is already broadcast and returns the reference.
    * If not, then broadcasts the model and returns a reference
+   * This is done to prevent the rebroadcasting of model during prediction and should
+   * only be mixed in with a model attribute.
    * @param sc SparkContext that will be used for the broadcast
    * @param modelToBc Model object to broadcast
    * @return the broadcast model