Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-21086 CrossValidator, TrainValidationSplit should preserve all models after fitting
  3. SPARK-21087

CrossValidator, TrainValidationSplit should collect all models when fitting: Scala API

    XMLWordPrintableJSON

    Details

    • Type: Sub-task
    • Status: Resolved
    • Priority: Major
    • Resolution: Fixed
    • Affects Version/s: 2.2.0
    • Fix Version/s: 2.3.0
    • Component/s: ML
    • Labels:
      None

      Description

      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

        Attachments

          Issue Links

            Activity

              People

              • Assignee:
                weichenxu123 Weichen Xu
                Reporter:
                josephkb Joseph K. Bradley
                Shepherd:
                Joseph K. Bradley
              • Votes:
                0 Vote for this issue
                Watchers:
                4 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved: