-
- Downloads
[SPARK-14386][ML] Changed spark.ml ensemble trees methods to return concrete types
## What changes were proposed in this pull request? In spark.ml, GBT and RandomForest expose the trait DecisionTreeModel in the trees method, but they should not since it is a private trait (and not ready to be made public). It will also be more useful to users if we return the concrete types. This PR: return concrete types The MIMA checks appear to be OK with this change. ## How was this patch tested? Existing unit tests Author: Joseph K. Bradley <joseph@databricks.com> Closes #12158 from jkbradley/hide-dtm.
Showing
- 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/classification/RandomForestClassifier.scala 3 additions, 3 deletions...ache/spark/ml/classification/RandomForestClassifier.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/regression/RandomForestRegressor.scala 3 additions, 2 deletions...rg/apache/spark/ml/regression/RandomForestRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/treeModels.scala 9 additions, 5 deletions.../src/main/scala/org/apache/spark/ml/tree/treeModels.scala
- mllib/src/test/scala/org/apache/spark/ml/tree/impl/TreeTests.scala 1 addition, 1 deletion.../test/scala/org/apache/spark/ml/tree/impl/TreeTests.scala
Loading
Please register or sign in to comment