From 82c1c5772817785709b0289f7d836beba812c791 Mon Sep 17 00:00:00 2001 From: Xiangrui Meng Date: Tue, 27 Oct 2015 23:41:42 -0700 Subject: [MINOR][ML] fix compile warns This fixes some compile time warnings. Author: Xiangrui Meng Closes #9319 from mengxr/mllib-compile-warn-20151027. --- .../main/scala/org/apache/spark/ml/regression/LinearRegression.scala | 2 +- .../org/apache/spark/ml/classification/LogisticRegressionSuite.scala | 3 ++- 2 files changed, 3 insertions(+), 2 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) || diff --git a/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala index 5186c4e2be..e0a795e5e0 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala @@ -17,6 +17,7 @@ package org.apache.spark.ml.classification +import scala.language.existentials import scala.util.Random import org.apache.spark.SparkFunSuite @@ -24,7 +25,7 @@ import org.apache.spark.ml.feature.Instance import org.apache.spark.ml.param.ParamsSuite import org.apache.spark.ml.util.MLTestingUtils import org.apache.spark.mllib.classification.LogisticRegressionSuite._ -import org.apache.spark.mllib.linalg.{Vectors, Vector} +import org.apache.spark.mllib.linalg.{Vector, Vectors} import org.apache.spark.mllib.regression.LabeledPoint import org.apache.spark.mllib.util.MLlibTestSparkContext import org.apache.spark.mllib.util.TestingUtils._ -- cgit v1.2.3