-
- Downloads
[SPARK-6684] [MLLIB] [ML] Add checkpointing to GBTs
Add checkpointing to GradientBoostedTrees, GBTClassifier, GBTRegressor CC: mengxr Author: Joseph K. Bradley <joseph@databricks.com> Closes #7804 from jkbradley/gbt-checkpoint3 and squashes the following commits: 3fbd7ba [Joseph K. Bradley] tiny fix b3e160c [Joseph K. Bradley] unset checkpoint dir after test 9cc3a04 [Joseph K. Bradley] added checkpointing to GBTs
Showing
- mllib/src/main/scala/org/apache/spark/mllib/clustering/LDAOptimizer.scala 1 addition, 0 deletions...cala/org/apache/spark/mllib/clustering/LDAOptimizer.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/GradientBoostedTrees.scala 25 additions, 23 deletions...la/org/apache/spark/mllib/tree/GradientBoostedTrees.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/BoostingStrategy.scala 2 additions, 1 deletion...che/spark/mllib/tree/configuration/BoostingStrategy.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/GBTClassifierSuite.scala 20 additions, 0 deletions...g/apache/spark/ml/classification/GBTClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/GBTRegressorSuite.scala 19 additions, 1 deletion...la/org/apache/spark/ml/regression/GBTRegressorSuite.scala
- mllib/src/test/scala/org/apache/spark/mllib/tree/GradientBoostedTreesSuite.scala 47 additions, 32 deletions...g/apache/spark/mllib/tree/GradientBoostedTreesSuite.scala
Loading
Please register or sign in to comment