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
