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

Trees and ensembles: More prediction functionality

    XMLWordPrintableJSON

    Details

    • Type: Improvement
    • Status: Resolved
    • Priority: Major
    • Resolution: Incomplete
    • Affects Version/s: None
    • Fix Version/s: None
    • Component/s: MLlib
    • Labels:

      Description

      DecisionTree and RandomForest currently predict the most likely label for classification and the mean for regression. Other info about predictions would be useful.

      For classification: estimated probability of each possible label
      For regression: variance of estimate

      RandomForest could also create aggregate predictions in multiple ways:

      • Predict mean or median value for regression.
      • Compute variance of estimates (across all trees) for both classification and regression.

        Attachments

          Issue Links

            Activity

              People

              • Assignee:
                Unassigned
                Reporter:
                josephkb Joseph K. Bradley
              • Votes:
                9 Vote for this issue
                Watchers:
                25 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved:

                  Time Tracking

                  Estimated:
                  Original Estimate - 72h
                  72h
                  Remaining:
                  Remaining Estimate - 72h
                  72h
                  Logged:
                  Time Spent - Not Specified
                  Not Specified