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

Make Logistic, Linear Regression Model evaluate() method public

    XMLWordPrintableJSON

Details

    • Improvement
    • Status: Resolved
    • Major
    • Resolution: Fixed
    • None
    • 2.0.0
    • ML
    • None

    Description

      The following method in LogisticRegressionModel is marked as private, which prevents users from creating a summary on any given data set. Check here.

      // TODO: decide on a good name before exposing to public API
      private[classification] def evaluate(dataset: DataFrame)
      : LogisticRegressionSummary = {
          new BinaryLogisticRegressionSummary(
              this.transform(dataset), 
              $(probabilityCol), 
              $(labelCol))
      }
      

      This method is definitely necessary to test model performance.
      By the way, the name evaluate is already pretty good for me.

      mengxr Could you check this ? Thx

      Attachments

        Activity

          People

            josephkb Joseph K. Bradley
            invkrh Hao Ren
            Votes:
            0 Vote for this issue
            Watchers:
            7 Start watching this issue

            Dates

              Created:
              Updated:
              Resolved: