org.apache.spark.ml.param.shared

Interface HasCollectSubModels

All Superinterfaces:

Identifiable, Params, java.io.Serializable

All Known Implementing Classes:

CrossValidator, TrainValidationSplit

public interface HasCollectSubModels
extends Params

Trait for shared param collectSubModels (default: false). This trait may be changed or removed between minor versions.

Method Summary

Modifier and Type	Method and Description
BooleanParam	collectSubModels() Param for whether to collect a list of sub-models trained during tuning.
boolean	getCollectSubModels()

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

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Method Detail

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()