[
https://issues.apache.org/jira/browse/SPARK-21088?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Joseph K. Bradley reassigned SPARK-21088:
-----------------------------------------
Assignee: Weichen Xu
> CrossValidator, TrainValidationSplit should collect all models when fitting:
> Python API
> ---------------------------------------------------------------------------------------
>
> Key: SPARK-21088
> URL: https://issues.apache.org/jira/browse/SPARK-21088
> Project: Spark
> Issue Type: Sub-task
> Components: ML, PySpark
> Affects Versions: 2.2.0
> Reporter: Joseph K. Bradley
> Assignee: Weichen Xu
> Priority: Major
>
> In pyspark:
> 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
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
