Skip to content

Commit

Permalink
fixed scala style error in NaiveBayes
Browse files Browse the repository at this point in the history
  • Loading branch information
leahmcguire committed Feb 27, 2015
1 parent 3891bf2 commit 5a4a534
Showing 1 changed file with 4 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,10 @@ object NaiveBayesModel extends Loader[NaiveBayesModel] {
def thisClassName = "org.apache.spark.mllib.classification.NaiveBayesModel"

/** Model data for model import/export */
case class Data(labels: Array[Double], pi: Array[Double], theta: Array[Array[Double]], modelType: String)
case class Data(labels: Array[Double],
pi: Array[Double],
theta: Array[Array[Double]],
modelType: String)

def save(sc: SparkContext, path: String, data: Data): Unit = {
val sqlContext = new SQLContext(sc)
Expand Down

0 comments on commit 5a4a534

Please sign in to comment.