ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[MINOR][ML] fix compile warns

Browse source 
This fixes some compile time warnings.

Author: Xiangrui Meng <meng@databricks.com>

Closes #9319 from mengxr/mllib-compile-warn-20151027.
This commit is contained in:
Xiangrui Meng 2015-10-27 23:41:42 -07:00
parent 826e1e304b
commit 82c1c57728
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala
View file

@ -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) ||

3
mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala
View file

@ -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._