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

Predicted Probability per training instance for Gradient Boosted Trees

    XMLWordPrintableJSON

Details

    • New Feature
    • Status: Resolved
    • Minor
    • Resolution: Fixed
    • None
    • 2.2.0
    • ML

    Description

      This function available for Logistic Regression, SVM etc. (model.setThreshold()) but not for GBT. In comparison to "gbm" package in R, where we can specify the distribution and get predicted probabilities or classes. I understand that this algorithm works with "Classification" and "Regression" algo's. Is there any way where in GBT we can get predicted probabilities or provide thresholds to the model?

      Attachments

        Issue Links

          Is contained by

          Umbrella - An overarching type made of sub-tasks SPARK-14047 GBT improvement umbrella

          • Major - Major loss of function.
          • Resolved
          links to

          Pull request #16441 [Github] Pull Request #16441 (imatiach-msft)

          Activity

            People

              imatiach Ilya Matiach
              partha_paxy_000777 Partha Talukder
              Joseph K. Bradley Joseph K. Bradley
              Votes:
              1 Vote for this issue
              Watchers:
              5 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: