GitHub user ludatabricks opened a pull request: https://github.com/apache/spark/pull/21044
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 http://spark.apache.org/contributing.html before opening a pull request. You can merge this pull request into a Git repository by running: $ git pull https://github.com/ludatabricks/spark-1 SPARK-9312 Alternatively you can review and apply these changes as the patch at: https://github.com/apache/spark/pull/21044.patch 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 <lu.wang@...> Date: 2018-04-11T19:08:22Z add rawPrediction as an output column; add numCLasses and numFeatures to OneVsRestModel ---- --- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org