aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2016-05-19 17:56:21 -0700
committerXiangrui Meng <meng@databricks.com>2016-05-19 17:56:21 -0700
commit664367781786df7ec52e39950dccd5a09681602c (patch)
treeed7e1da4e85d76ad7d3a803d6019b72b883aecea /python/pyspark/ml
parentf8107c7846c9fcabbe2579867574305c7f2028e7 (diff)
downloadspark-664367781786df7ec52e39950dccd5a09681602c.tar.gz
spark-664367781786df7ec52e39950dccd5a09681602c.tar.bz2
spark-664367781786df7ec52e39950dccd5a09681602c.zip
[MINOR][ML][PYSPARK] ml.evaluation Scala and Python API sync
## What changes were proposed in this pull request? ```ml.evaluation``` Scala and Python API sync. ## How was this patch tested? Only API docs change, no new tests. Author: Yanbo Liang <ybliang8@gmail.com> Closes #13195 from yanboliang/evaluation-doc.
Diffstat (limited to 'python/pyspark/ml')
-rw-r--r--python/pyspark/ml/evaluation.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/python/pyspark/ml/evaluation.py b/python/pyspark/ml/evaluation.py
index fc9099b7ec..16029dc348 100644
--- a/python/pyspark/ml/evaluation.py
+++ b/python/pyspark/ml/evaluation.py
@@ -193,9 +193,6 @@ class RegressionEvaluator(JavaEvaluator, HasLabelCol, HasPredictionCol):
.. versionadded:: 1.4.0
"""
- # Because we will maximize evaluation value (ref: `CrossValidator`),
- # when we evaluate a metric that is needed to minimize (e.g., `"rmse"`, `"mse"`, `"mae"`),
- # we take and output the negative of this metric.
metricName = Param(Params._dummy(), "metricName",
"""metric name in evaluation - one of:
rmse - root mean squared error (default)
@@ -270,7 +267,7 @@ class MulticlassClassificationEvaluator(JavaEvaluator, HasLabelCol, HasPredictio
"""
metricName = Param(Params._dummy(), "metricName",
"metric name in evaluation "
- "(f1|precision|recall|weightedPrecision|weightedRecall)",
+ "(f1|precision|recall|weightedPrecision|weightedRecall|accuracy)",
typeConverter=TypeConverters.toString)
@keyword_only