Skip to content

Commit b836bac

Browse files
viiryayhuai
authored andcommitted
[HOTFIX] Hotfix branch-1.4 building by removing avgMetrics in CrossValidatorSuite
Ref. #6905 ping yhuai Author: Liang-Chi Hsieh <viirya@gmail.com> Closes #6929 from viirya/hot_fix_cv_test and squashes the following commits: b1aec53 [Liang-Chi Hsieh] Hotfix branch-1.4 by removing avgMetrics in CrossValidatorSuite.
1 parent 2a7ea31 commit b836bac

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

mllib/src/test/scala/org/apache/spark/ml/tuning/CrossValidatorSuite.scala

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,14 +78,12 @@ class CrossValidatorSuite extends SparkFunSuite with MLlibTestSparkContext {
7878
val parent = cvModel.bestModel.parent.asInstanceOf[LinearRegression]
7979
assert(parent.getRegParam === 0.001)
8080
assert(parent.getMaxIter === 10)
81-
assert(cvModel.avgMetrics.length === lrParamMaps.length)
8281

8382
eval.setMetricName("r2")
8483
val cvModel2 = cv.fit(dataset)
8584
val parent2 = cvModel2.bestModel.parent.asInstanceOf[LinearRegression]
8685
assert(parent2.getRegParam === 0.001)
8786
assert(parent2.getMaxIter === 10)
88-
assert(cvModel2.avgMetrics.length === lrParamMaps.length)
8987
}
9088

9189
test("validateParams should check estimatorParamMaps") {

0 commit comments

Comments
 (0)