diff options
author | Xiangrui Meng <meng@databricks.com> | 2015-10-27 23:41:42 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-10-27 23:41:42 -0700 |
commit | 82c1c5772817785709b0289f7d836beba812c791 (patch) | |
tree | ac26aca2b0ce68333d7ca4f8045c5596fa65eddd /mllib/src/main/scala/org | |
parent | 826e1e304b57abbc56b8b7ffd663d53942ab3c7c (diff) | |
download | spark-82c1c5772817785709b0289f7d836beba812c791.tar.gz spark-82c1c5772817785709b0289f7d836beba812c791.tar.bz2 spark-82c1c5772817785709b0289f7d836beba812c791.zip |
[MINOR][ML] fix compile warns
This fixes some compile time warnings.
Author: Xiangrui Meng <meng@databricks.com>
Closes #9319 from mengxr/mllib-compile-warn-20151027.
Diffstat (limited to 'mllib/src/main/scala/org')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala index 573a61a6ea..c3ee8b3bc1 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala @@ -145,7 +145,7 @@ class LinearRegression(override val uid: String) // Extract the number of features before deciding optimization solver. val numFeatures = dataset.select(col($(featuresCol))).limit(1).map { case Row(features: Vector) => features.size - }.toArray()(0) + }.first() val w = if ($(weightCol).isEmpty) lit(1.0) else col($(weightCol)) if (($(solver) == "auto" && $(elasticNetParam) == 0.0 && numFeatures <= 4096) || |