Interface HasCollectSubModels

All Superinterfaces:
Identifiable, Params, Serializable, scala.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 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.
    • getCollectSubModels

      boolean getCollectSubModels()