Skip to content
Snippets Groups Projects
Commit 66436778 authored by Yanbo Liang's avatar Yanbo Liang Committed by Xiangrui Meng
Browse files

[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.
parent f8107c78
No related branches found
No related tags found
No related merge requests found
......@@ -28,7 +28,7 @@ import org.apache.spark.sql.types.DoubleType
/**
* :: Experimental ::
* Evaluator for multiclass classification, which expects two input columns: score and label.
* Evaluator for multiclass classification, which expects two input columns: prediction and label.
*/
@Since("1.5.0")
@Experimental
......
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment