[
https://issues.apache.org/jira/browse/SPARK-21087?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Joseph K. Bradley reassigned SPARK-21087:
-----------------------------------------
Assignee: Weichen Xu
> CrossValidator, TrainValidationSplit should collect all models when fitting:
> Scala API
> --------------------------------------------------------------------------------------
>
> Key: SPARK-21087
> URL: https://issues.apache.org/jira/browse/SPARK-21087
> Project: Spark
> Issue Type: Sub-task
> Components: ML
> Affects Versions: 2.2.0
> Reporter: Joseph K. Bradley
> Assignee: Weichen Xu
> Priority: Major
>
> We add a parameter whether to collect the full model list when
> CrossValidator/TrainValidationSplit training (Default is NOT, avoid the
> change cause OOM)
> Add a method in CrossValidatorModel/TrainValidationSplitModel, allow user to
> get the model list
> CrossValidatorModelWriter add a “option”, allow user to control whether to
> persist the model list to disk.
> Note: when persisting the model list, use indices as the sub-model path
--
This message was sent by Atlassian JIRA
(v6.4.14#64029)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
