Skip to content

Commit 283b04a

Browse files
author
DB Tsai
committed
more test fix
1 parent 6faec8a commit 283b04a

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mllib/src/test/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifierSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,11 @@
1818
package org.apache.spark.ml.classification
1919

2020
import org.apache.spark.SparkFunSuite
21+
import org.apache.spark.ml.classification.LogisticRegressionSuite._
2122
import org.apache.spark.ml.linalg.{Vector, Vectors}
2223
import org.apache.spark.ml.util.DefaultReadWriteTest
2324
import org.apache.spark.ml.util.MLTestingUtils
2425
import org.apache.spark.ml.util.TestingUtils._
25-
import org.apache.spark.mllib.classification.LogisticRegressionSuite._
2626
import org.apache.spark.mllib.classification.LogisticRegressionWithLBFGS
2727
import org.apache.spark.mllib.evaluation.MulticlassMetrics
2828
import org.apache.spark.mllib.util.MLlibTestSparkContext

mllib/src/test/scala/org/apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ class RegressionEvaluatorSuite
4444
*/
4545
val dataset = sqlContext.createDataFrame(
4646
sc.parallelize(LinearDataGenerator.generateLinearInput(
47-
6.3, Array(4.7, 7.2), Array(0.9, -1.3), Array(0.7, 1.2), 100, 42, 0.1), 2))
47+
6.3, Array(4.7, 7.2), Array(0.9, -1.3), Array(0.7, 1.2), 100, 42, 0.1), 2).map(_.asML))
4848

4949
/**
5050
* Using the following R code to load the data, train the model and evaluate metrics.

0 commit comments

Comments
 (0)