Skip to content

Commit b349de7

Browse files
committed
ml.evaluation Scala and Python API sync
1 parent e2ec32d commit b349de7

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

mllib/src/main/scala/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluator.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ import org.apache.spark.sql.types.DoubleType
2828

2929
/**
3030
* :: Experimental ::
31-
* Evaluator for multiclass classification, which expects two input columns: score and label.
31+
* Evaluator for multiclass classification, which expects two input columns: prediction and label.
3232
*/
3333
@Since("1.5.0")
3434
@Experimental

python/pyspark/ml/evaluation.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -193,9 +193,7 @@ class RegressionEvaluator(JavaEvaluator, HasLabelCol, HasPredictionCol):
193193
194194
.. versionadded:: 1.4.0
195195
"""
196-
# Because we will maximize evaluation value (ref: `CrossValidator`),
197-
# when we evaluate a metric that is needed to minimize (e.g., `"rmse"`, `"mse"`, `"mae"`),
198-
# we take and output the negative of this metric.
196+
199197
metricName = Param(Params._dummy(), "metricName",
200198
"""metric name in evaluation - one of:
201199
rmse - root mean squared error (default)
@@ -270,7 +268,7 @@ class MulticlassClassificationEvaluator(JavaEvaluator, HasLabelCol, HasPredictio
270268
"""
271269
metricName = Param(Params._dummy(), "metricName",
272270
"metric name in evaluation "
273-
"(f1|precision|recall|weightedPrecision|weightedRecall)",
271+
"(f1|precision|recall|weightedPrecision|weightedRecall|accuracy)",
274272
typeConverter=TypeConverters.toString)
275273

276274
@keyword_only

0 commit comments

Comments
 (0)