-
- Downloads
[SPARK-12379][ML][MLLIB] Copy GBT implementation to spark.ml
Currently, GBTs in spark.ml wrap the implementation in spark.mllib. This is preventing several improvements to GBTs in spark.ml, so we need to move the implementation to ml and use spark.ml decision trees in the implementation. At first, we should make minimal changes to the implementation. Performance testing should be done to ensure there were no regressions. Performance testing results are [here](https://docs.google.com/document/d/1dYd2mnfGdUKkQ3vZe2BpzsTnI5IrpSLQ-NNKDZhUkgw/edit?usp=sharing) Author: sethah <seth.hendrickson16@gmail.com> Closes #10607 from sethah/SPARK-12379.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/DecisionTreeClassifier.scala 8 additions, 0 deletions...ache/spark/ml/classification/DecisionTreeClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/GBTClassifier.scala 3 additions, 4 deletions...la/org/apache/spark/ml/classification/GBTClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/DecisionTreeRegressor.scala 8 additions, 0 deletions...rg/apache/spark/ml/regression/DecisionTreeRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala 3 additions, 4 deletions...n/scala/org/apache/spark/ml/regression/GBTRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/impl/GradientBoostedTrees.scala 277 additions, 0 deletions.../org/apache/spark/ml/tree/impl/GradientBoostedTrees.scala
- mllib/src/main/scala/org/apache/spark/mllib/impl/PeriodicRDDCheckpointer.scala 1 addition, 1 deletion...org/apache/spark/mllib/impl/PeriodicRDDCheckpointer.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/BoostingStrategy.scala 1 addition, 1 deletion...che/spark/mllib/tree/configuration/BoostingStrategy.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala 1 addition, 1 deletion.../org/apache/spark/mllib/tree/configuration/Strategy.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/AbsoluteError.scala 1 addition, 1 deletion...cala/org/apache/spark/mllib/tree/loss/AbsoluteError.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/LogLoss.scala 1 addition, 1 deletion...main/scala/org/apache/spark/mllib/tree/loss/LogLoss.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/Loss.scala 1 addition, 1 deletion...rc/main/scala/org/apache/spark/mllib/tree/loss/Loss.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/SquaredError.scala 1 addition, 1 deletion...scala/org/apache/spark/mllib/tree/loss/SquaredError.scala
Loading
Please register or sign in to comment