-
- Downloads
[SPARK-6025] [MLlib] Add helper method evaluateEachIteration to extract learning curve
Added evaluateEachIteration to allow the user to manually extract the error for each iteration of GradientBoosting. The internal optimisation can be dealt with later. Author: MechCoder <manojkumarsivaraj334@gmail.com> Closes #4906 from MechCoder/spark-6025 and squashes the following commits: 67146ab [MechCoder] Minor 352001f [MechCoder] Minor 6e8aa10 [MechCoder] Made the following changes Used mapPartition instead of map Refactored computeError and unpersisted broadcast variables bc99ac6 [MechCoder] Refactor the method and stuff dbda033 [MechCoder] [SPARK-6025] Add helper method evaluateEachIteration to extract learning curve
Showing
- docs/mllib-ensembles.md 2 additions, 2 deletionsdocs/mllib-ensembles.md
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/AbsoluteError.scala 4 additions, 13 deletions...cala/org/apache/spark/mllib/tree/loss/AbsoluteError.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/LogLoss.scala 5 additions, 15 deletions...main/scala/org/apache/spark/mllib/tree/loss/LogLoss.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/loss/Loss.scala 13 additions, 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 4 additions, 13 deletions...scala/org/apache/spark/mllib/tree/loss/SquaredError.scala
- mllib/src/main/scala/org/apache/spark/mllib/tree/model/treeEnsembleModels.scala 54 additions, 0 deletions...rg/apache/spark/mllib/tree/model/treeEnsembleModels.scala
- mllib/src/test/scala/org/apache/spark/mllib/tree/GradientBoostedTreesSuite.scala 14 additions, 2 deletions...g/apache/spark/mllib/tree/GradientBoostedTreesSuite.scala
Loading
Please register or sign in to comment