GitHub user pralabhkumar opened a pull request:
https://github.com/apache/spark/pull/18118
SPARK-20199 : Provided featureSubsetStrategy to GBTClassifier
## What changes were proposed in this pull request?
(Provided featureSubset Strategy to GBTClassifier
a) Moved featureSubsetStrategy to TreeEnsembleParams
b) Changed GBTClassifier to pass featureSubsetStrategy
val firstTreeModel = firstTree.train(input, treeStrategy,
featureSubsetStrategy))
## How was this patch tested?
a) Tested GradientBoostedTreeClassifierExample by adding
.setFeatureSubsetStrategy with GBTClassifier
(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/pralabhkumar/spark develop
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/18118.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 #18118
----
commit b0444fa75f4cc33a0c35cf88664a89a1c425e7a1
Author: Pralabh Kumar <pralabhku...@gmail.com>
Date: 2017-05-26T07:16:32Z
SPARK-20199 : Provided featureSubsetStrategy to GBTClassifier
----
---
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
