GitHub user ludatabricks opened a pull request:

    Add RawPrediction, numClasses, and numFeatures for OneVsRestModel

    add RawPrediction as output column 
    add numClasses and numFeatures to OneVsRestModel
    ## What changes were proposed in this pull request?
    - Add two val numClasses and numFeatures in OneVsRestModel so that we can 
inherit from Classifier in the future
    - Add rawPrediction output column in transform, the prediction label in 
calculated by the rawPrediciton like raw2prediction
    ## How was this patch tested?
    (Please explain how this patch was tested. E.g. unit tests, integration 
tests, manual tests)
    (If this patch involves UI changes, please attach a screenshot; otherwise, 
remove this)
    Please review before opening a 
pull request.

You can merge this pull request into a Git repository by running:

    $ git pull SPARK-9312

Alternatively you can review and apply these changes as the patch at:

To close this pull request, make a commit to your master/trunk branch
with (at least) the following in the commit message:

    This closes #21044
commit 0cfc20a3637c06071e6fe48ca5db4834b34c889e
Author: Lu WANG <>
Date:   2018-04-11T19:08:22Z

    add rawPrediction as an output column;
    add numCLasses and numFeatures to OneVsRestModel



To unsubscribe, e-mail:
For additional commands, e-mail:

Reply via email to