-
- Downloads
[SPARK-9715] [ML] Store numFeatures in all ML PredictionModel types
All prediction models should store `numFeatures` indicating the number of features the model was trained on. Default value of -1 added for backwards compatibility. Author: sethah <seth.hendrickson16@gmail.com> Closes #8675 from sethah/SPARK-9715.
Showing
- examples/src/main/java/org/apache/spark/examples/ml/JavaDeveloperApiExample.java 5 additions, 0 deletions...org/apache/spark/examples/ml/JavaDeveloperApiExample.java
- examples/src/main/scala/org/apache/spark/examples/ml/DeveloperApiExample.scala 3 additions, 0 deletions...la/org/apache/spark/examples/ml/DeveloperApiExample.scala
- mllib/src/main/scala/org/apache/spark/ml/Predictor.scala 5 additions, 1 deletionmllib/src/main/scala/org/apache/spark/ml/Predictor.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/DecisionTreeClassifier.scala 8 additions, 5 deletions...ache/spark/ml/classification/DecisionTreeClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/GBTClassifier.scala 19 additions, 7 deletions...la/org/apache/spark/ml/classification/GBTClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 2 additions, 0 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala 2 additions, 0 deletions...rk/ml/classification/MultilayerPerceptronClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/NaiveBayes.scala 2 additions, 0 deletions...scala/org/apache/spark/ml/classification/NaiveBayes.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/RandomForestClassifier.scala 4 additions, 4 deletions...ache/spark/ml/classification/RandomForestClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/DecisionTreeRegressor.scala 8 additions, 5 deletions...rg/apache/spark/ml/regression/DecisionTreeRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala 18 additions, 6 deletions...n/scala/org/apache/spark/ml/regression/GBTRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala 2 additions, 0 deletions...ala/org/apache/spark/ml/regression/LinearRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala 4 additions, 3 deletions...rg/apache/spark/ml/regression/RandomForestRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala 10 additions, 4 deletions...in/scala/org/apache/spark/ml/tree/impl/RandomForest.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala 3 additions, 1 deletion...spark/ml/classification/DecisionTreeClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/GBTClassifierSuite.scala 7 additions, 4 deletions...g/apache/spark/ml/classification/GBTClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala 2 additions, 0 deletions...che/spark/ml/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifierSuite.scala 3 additions, 1 deletion.../classification/MultilayerPerceptronClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/ProbabilisticClassifierSuite.scala 4 additions, 2 deletions...park/ml/classification/ProbabilisticClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/RandomForestClassifierSuite.scala 5 additions, 3 deletions...spark/ml/classification/RandomForestClassifierSuite.scala
Loading
Please register or sign in to comment