GitHub user imatiach-msft opened a pull request:
https://github.com/apache/spark/pull/16441
[SPARK-14975][ML][WIP] Fixed GBTClassifier to predict probability per
training instance and fixed interfaces
## What changes were proposed in this pull request?
For all of the classifiers in MLLib we can predict probabilities except for
GBTClassifier.
Also, all classifiers inherit from ProbabilisticClassifier but
GBTClassifier strangely inherits from Predictor, which is a bug.
This change corrects the interface and adds the ability for the classifier
to give a probabilities vector.
## How was this patch tested?
The basic ML tests were run after making the changes. I've marked this as
WIP as I need to add more tests.
You can merge this pull request into a Git repository by running:
$ git pull https://github.com/imatiach-msft/spark ilmat/fix-GBT
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/16441.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 #16441
----
commit 63a9574a0858ed9e4c27a4b698cb50d2475afc0b
Author: Ilya Matiach <il...@microsoft.com>
Date: 2016-12-30T20:15:12Z
[SPARK-14975][ML][WIP] Fixed GBTClassifier to predict probability per
training instance and fixed interfaces
commit 4468891cd83760a2f97ef257ef176a34bc79e5cd
Author: Ilya Matiach <il...@microsoft.com>
Date: 2016-12-30T20:20:43Z
Fixed scala style empty line
----
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
