diff options
| author | Joseph K. Bradley <joseph@databricks.com> | 2015-05-06 16:15:51 -0700 |
|---|---|---|
| committer | Xiangrui Meng <meng@databricks.com> | 2015-05-06 16:15:59 -0700 |
| commit | b681b9312efa3914db980705e401e58082754f1d (patch) | |
| tree | 5a11ceea3286634eb11bcf87273b2647d081b0aa /project | |
| parent | 14bcb84e84be9d9cc847b4b9d5d803f5ecb89bc5 (diff) | |
| download | spark-b681b9312efa3914db980705e401e58082754f1d.tar.gz spark-b681b9312efa3914db980705e401e58082754f1d.tar.bz2 spark-b681b9312efa3914db980705e401e58082754f1d.zip | |
[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.
(cherry picked from commit 1ad04dae038673a448f529c39b17817b78d6acd0)
Signed-off-by: Xiangrui Meng <meng@databricks.com>
Diffstat (limited to 'project')
0 files changed, 0 insertions, 0 deletions