Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-7127

Broadcast spark.ml tree ensemble models for predict

    XMLWordPrintableJSON

Details

    • Improvement
    • Status: Resolved
    • Minor
    • Resolution: Fixed
    • 1.4.0
    • 1.5.0
    • ML
    • None

    Description

      GBTRegressor/Classifier and RandomForestRegressor/Classifier should broadcast models and then predict. This will mean overriding transform().

      Note: Try to reduce duplicated code via the TreeEnsembleModel abstraction.

      Attachments

        Issue Links

          is blocked by

          Sub-task - The sub-task of the issue SPARK-6113 Stabilize DecisionTree and ensembles APIs

          • Critical - Crashes, loss of data, severe memory leak.
          • Resolved
          links to

          Pull request #6300 [Github] Pull Request #6300 (BryanCutler)

          Activity

            People

              bryanc Bryan Cutler
              josephkb Joseph K. Bradley
              Joseph K. Bradley Joseph K. Bradley
              Votes:
              0 Vote for this issue
              Watchers:
              4 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: