We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2a7ea31 commit b1aec53Copy full SHA for b1aec53
1 file changed
mllib/src/test/scala/org/apache/spark/ml/tuning/CrossValidatorSuite.scala
@@ -78,14 +78,12 @@ class CrossValidatorSuite extends SparkFunSuite with MLlibTestSparkContext {
78
val parent = cvModel.bestModel.parent.asInstanceOf[LinearRegression]
79
assert(parent.getRegParam === 0.001)
80
assert(parent.getMaxIter === 10)
81
- assert(cvModel.avgMetrics.length === lrParamMaps.length)
82
83
eval.setMetricName("r2")
84
val cvModel2 = cv.fit(dataset)
85
val parent2 = cvModel2.bestModel.parent.asInstanceOf[LinearRegression]
86
assert(parent2.getRegParam === 0.001)
87
assert(parent2.getMaxIter === 10)
88
- assert(cvModel2.avgMetrics.length === lrParamMaps.length)
89
}
90
91
test("validateParams should check estimatorParamMaps") {
0 commit comments