Dongjoon Hyun created SPARK-14907:
-------------------------------------
Summary: Use repartition in GLMRegressionModel.save
Key: SPARK-14907
URL: https://issues.apache.org/jira/browse/SPARK-14907
Project: Spark
Issue Type: Improvement
Components: MLlib
Reporter: Dongjoon Hyun
Priority: Trivial
This issue changes `GLMRegressionModel.save` function like the following code
that is similar to other algorithms' parquet write.
{code}
- val dataRDD: DataFrame = sc.parallelize(Seq(data), 1).toDF()
- // TODO: repartition with 1 partition after SPARK-5532 gets fixed
- dataRDD.write.parquet(Loader.dataPath(path))
+
sqlContext.createDataFrame(Seq(data)).repartition(1).write.parquet(Loader.dataPath(path))
{code}
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
