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

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

    XMLWordPrintableJSON

Details

    • Sub-task
    • Status: Resolved
    • Major
    • Resolution: Fixed
    • 2.2.0
    • 2.4.0
    • ML, PySpark
    • None

    Description

      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

      Attachments

        Issue Links

          Activity

            People

              weichenxu123 Weichen Xu
              josephkb Joseph K. Bradley
              Votes:
              0 Vote for this issue
              Watchers:
              5 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: