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

Predicted Probability per training instance for Gradient Boosted Trees

    XMLWordPrintableJSON

    Details

    • Type: New Feature
    • Status: Resolved
    • Priority: Minor
    • Resolution: Fixed
    • Affects Version/s: None
    • Fix Version/s: 2.2.0
    • Component/s: ML
    • Labels:

      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

            Activity

              People

              • Assignee:
                imatiach Ilya Matiach
                Reporter:
                partha_paxy_000777 Partha Talukder
                Shepherd:
                Joseph K. Bradley
              • Votes:
                1 Vote for this issue
                Watchers:
                6 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved: