-
- Downloads
[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.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/GBTClassifier.scala 69 additions, 41 deletions...la/org/apache/spark/ml/classification/GBTClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/RandomForestClassifier.scala 1 addition, 1 deletion...ache/spark/ml/classification/RandomForestClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala 71 additions, 43 deletions...n/scala/org/apache/spark/ml/regression/GBTRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala 1 addition, 1 deletion...rg/apache/spark/ml/regression/RandomForestRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/treeModels.scala 15 additions, 10 deletions.../src/main/scala/org/apache/spark/ml/tree/treeModels.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/treeParams.scala 72 additions, 1 deletion.../src/main/scala/org/apache/spark/ml/tree/treeParams.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/GBTClassifierSuite.scala 17 additions, 20 deletions...g/apache/spark/ml/classification/GBTClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/GBTRegressorSuite.scala 16 additions, 20 deletions...la/org/apache/spark/ml/regression/GBTRegressorSuite.scala
Loading
Please register or sign in to comment