diff options
author | sethah <seth.hendrickson16@gmail.com> | 2016-03-23 15:08:47 -0700 |
---|---|---|
committer | Joseph K. Bradley <joseph@databricks.com> | 2016-03-23 15:08:47 -0700 |
commit | 69bc2c17f1ca047d4915a4791b624d60c5943dc8 (patch) | |
tree | f59f8e0aae9421eab43bd2509b32e222ea104fa4 /core/src/test/scala/org/apache | |
parent | 5dfc01976bb0d72489620b4f32cc12d620bb6260 (diff) | |
download | spark-69bc2c17f1ca047d4915a4791b624d60c5943dc8.tar.gz spark-69bc2c17f1ca047d4915a4791b624d60c5943dc8.tar.bz2 spark-69bc2c17f1ca047d4915a4791b624d60c5943dc8.zip |
[SPARK-13952][ML] Add random seed to GBT
## What changes were proposed in this pull request?
`GBTClassifier` and `GBTRegressor` should use random seed for reproducible results. Because of the nature of current unit tests, which compare GBTs in ML and GBTs in MLlib for equality, I also added a random seed to MLlib GBT algorithm. I made alternate constructors in `mllib.tree.GradientBoostedTrees` to accept a random seed, but left them as private so as to not change the API unnecessarily.
## How was this patch tested?
Existing unit tests verify that functionality did not change. Other ML algorithms do not seem to have unit tests that directly test the functionality of random seeding, but reproducibility with seeding for GBTs is effectively verified in existing tests. I can add more tests if needed.
Author: sethah <seth.hendrickson16@gmail.com>
Closes #11903 from sethah/SPARK-13952.
Diffstat (limited to 'core/src/test/scala/org/apache')
0 files changed, 0 insertions, 0 deletions