-
- Downloads
[SPARK-11861][ML] Add feature importances for decision trees
This patch adds an API entry point for single decision tree feature importances. Author: sethah <seth.hendrickson16@gmail.com> Closes #9912 from sethah/SPARK-11861.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/DecisionTreeClassifier.scala 19 additions, 0 deletions...ache/spark/ml/classification/DecisionTreeClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/RandomForestClassifier.scala 2 additions, 2 deletions...ache/spark/ml/classification/RandomForestClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/DecisionTreeRegressor.scala 19 additions, 0 deletions...rg/apache/spark/ml/regression/DecisionTreeRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala 2 additions, 2 deletions...rg/apache/spark/ml/regression/RandomForestRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala 24 additions, 6 deletions...in/scala/org/apache/spark/ml/tree/impl/RandomForest.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala 21 additions, 0 deletions...spark/ml/classification/DecisionTreeClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/RandomForestClassifierSuite.scala 3 additions, 7 deletions...spark/ml/classification/RandomForestClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/impl/TreeTests.scala 13 additions, 0 deletions...b/src/test/scala/org/apache/spark/ml/impl/TreeTests.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/DecisionTreeRegressorSuite.scala 20 additions, 0 deletions...ache/spark/ml/regression/DecisionTreeRegressorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/RandomForestRegressorSuite.scala 3 additions, 10 deletions...ache/spark/ml/regression/RandomForestRegressorSuite.scala
Loading
Please register or sign in to comment