diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2016-04-13 11:31:10 -0700 |
---|---|---|
committer | Joseph K. Bradley <joseph@databricks.com> | 2016-04-13 11:31:10 -0700 |
commit | f9d578eaa107d8e8503c1563a2b3990c85104298 (patch) | |
tree | c4410c020f61d9c48780eb0f108be250d254f42f /repl/scala-2.11/src/main/scala | |
parent | 7d2ed8cc030f3d84fea47fded072c320c3d87ca7 (diff) | |
download | spark-f9d578eaa107d8e8503c1563a2b3990c85104298.tar.gz spark-f9d578eaa107d8e8503c1563a2b3990c85104298.tar.bz2 spark-f9d578eaa107d8e8503c1563a2b3990c85104298.zip |
[SPARK-13783][ML] Model export/import for spark.ml: GBTs
## What changes were proposed in this pull request?
* Added save/load for ```GBTClassifier/GBTClassificationModel/GBTRegressor/GBTRegressionModel```.
* Meanwhile, I modified ```EnsembleModelReadWrite.saveImpl/loadImpl``` to support save/load ```treeWeights```.
## How was this patch tested?
Adds standard unit tests for GBT save/load.
cc jkbradley GayathriMurali
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #12230 from yanboliang/spark-13783.
Diffstat (limited to 'repl/scala-2.11/src/main/scala')
0 files changed, 0 insertions, 0 deletions