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

Add missing APIs to Python GBTClassifier

    XMLWordPrintableJSON

Details

    • Improvement
    • Status: Resolved
    • Minor
    • Resolution: Fixed
    • 2.3.0
    • 2.4.0
    • ML, PySpark

    Description

      GBTClassifier is missing featureSubsetStrategy.  This should be moved to TreeEnsembleParams, as in Scala, and it will then be part of GBTs.

      GBTClassificationModel is missing numClasses. It should inherit from JavaClassificationModel instead of prediction model which will give it this param.

      Attachments

        Issue Links

          Activity

            People

              huaxingao Huaxin Gao
              bryanc Bryan Cutler
              Votes:
              0 Vote for this issue
              Watchers:
              4 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: