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

expose multiclass scores in LogisticRegression

Attach filesAttach ScreenshotVotersWatch issueWatchersCreate sub-taskLinkCloneUpdate Comment AuthorReplace String in CommentUpdate Comment VisibilityDelete Comments
    XMLWordPrintableJSON

    Details

      Description

      In SPARK-2309, DB Tsai implemented multinomial logistic regression for mllib. It would be great if users could get the score for each class directly, since they are already computed. It doesn't look like this would be a terribly complicated thing to do and I've sketched out what it could look like: https://gist.github.com/Krimit/f61b5e3be2b1e380d1ca

      We could even move the score methods further up into GeneralizedLinearModel, since in the binary case we could still just return a 2-element vector of the scores (or margin in SVM's) for each of the 2 class, so algorithms lacking multiclass support at the moment can still support this.

        Attachments

          Activity

            People

            • Assignee:
              Unassigned
              Reporter:
              akrim Asher Krim

              Dates

              • Created:
                Updated:
                Resolved:

                Issue deployment