Github user smurching commented on a diff in the pull request:
https://github.com/apache/spark/pull/19208#discussion_r139557219
--- Diff:
mllib/src/main/scala/org/apache/spark/ml/tuning/CrossValidator.scala ---
@@ -117,6 +123,12 @@ class CrossValidator @Since("1.2.0") (@Since("1.4.0")
override val uid: String)
instr.logParams(numFolds, seed, parallelism)
logTuningParams(instr)
+ val collectSubModelsParam = $(collectSubModels)
+
+ var subModels: Array[Array[Model[_]]] = if (collectSubModelsParam) {
--- End diff --
Perhaps use an `Option[Array[Model[_]]]` instead of setting `subModels` to
null?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
