-
- Downloads
[SPARK-5995] [ML] Make Prediction dev API public
Changes: * Update protected prediction methods, following design doc. **<--most interesting change** * Changed abstract classes for Estimator and Model to be public. Added DeveloperApi tag. (I kept the traits for Estimator/Model Params private.) * Changed ProbabilisticClassificationModel method names to use probability instead of probabilities. CC: mengxr shivaram etrain Author: Joseph K. Bradley <joseph@databricks.com> Closes #5913 from jkbradley/public-dev-api and squashes the following commits: e9aa0ea [Joseph K. Bradley] moved findMax to DenseVector and renamed to argmax. fixed bug for vector of length 0 15b9957 [Joseph K. Bradley] renamed probabilities to probability in method names 5cda84d [Joseph K. Bradley] regenerated sharedParams 7d1877a [Joseph K. Bradley] Made spark.ml prediction abstractions public. Organized their prediction methods for efficient computation of multiple output columns.
Showing
- mllib/src/main/scala/org/apache/spark/ml/Predictor.scala 12 additions, 38 deletionsmllib/src/main/scala/org/apache/spark/ml/Predictor.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/Classifier.scala 38 additions, 72 deletions...scala/org/apache/spark/ml/classification/Classifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/DecisionTreeClassifier.scala 2 additions, 3 deletions...ache/spark/ml/classification/DecisionTreeClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/GBTClassifier.scala 2 additions, 3 deletions...la/org/apache/spark/ml/classification/GBTClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 33 additions, 67 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala 68 additions, 32 deletions...che/spark/ml/classification/ProbabilisticClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/RandomForestClassifier.scala 2 additions, 3 deletions...ache/spark/ml/classification/RandomForestClassifier.scala
- mllib/src/main/scala/org/apache/spark/ml/param/shared/SharedParamsCodeGen.scala 4 additions, 2 deletions...rg/apache/spark/ml/param/shared/SharedParamsCodeGen.scala
- mllib/src/main/scala/org/apache/spark/ml/param/shared/sharedParams.scala 2 additions, 2 deletions...scala/org/apache/spark/ml/param/shared/sharedParams.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/DecisionTreeRegressor.scala 2 additions, 3 deletions...rg/apache/spark/ml/regression/DecisionTreeRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala 2 additions, 3 deletions...n/scala/org/apache/spark/ml/regression/GBTRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala 3 additions, 2 deletions...ala/org/apache/spark/ml/regression/LinearRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala 2 additions, 3 deletions...rg/apache/spark/ml/regression/RandomForestRegressor.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/Regressor.scala 10 additions, 32 deletions...main/scala/org/apache/spark/ml/regression/Regressor.scala
- mllib/src/main/scala/org/apache/spark/ml/tree/treeParams.scala 2 additions, 2 deletions.../src/main/scala/org/apache/spark/ml/tree/treeParams.scala
- mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala 22 additions, 0 deletions...rc/main/scala/org/apache/spark/mllib/linalg/Vectors.scala
Loading
Please register or sign in to comment