Package org.apache.spark.ml.param.shared
Interface HasCollectSubModels
- All Superinterfaces:
Identifiable
,Params
,Serializable
- All Known Implementing Classes:
CrossValidator
,TrainValidationSplit
Trait for shared param collectSubModels (default: false). This trait may be changed or
removed between minor versions.
-
Method Summary
Modifier and TypeMethodDescriptionParam for whether to collect a list of sub-models trained during tuning.boolean
Methods inherited from interface org.apache.spark.ml.util.Identifiable
toString, uid
Methods inherited from interface org.apache.spark.ml.param.Params
clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
-
Method Details
-
collectSubModels
BooleanParam collectSubModels()Param for whether to collect a list of sub-models trained during tuning. If set to false, then only the single best sub-model will be available after fitting. If set to true, then all sub-models will be available. Warning: For large models, collecting all sub-models can cause OOMs on the Spark driver.- Returns:
- (undocumented)
-
getCollectSubModels
boolean getCollectSubModels()
-