aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--python/pyspark/ml/evaluation.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/ml/evaluation.py b/python/pyspark/ml/evaluation.py
index 23c37167b3..d8ddb78c6d 100644
--- a/python/pyspark/ml/evaluation.py
+++ b/python/pyspark/ml/evaluation.py
@@ -205,7 +205,7 @@ class RegressionEvaluator(JavaEvaluator, HasLabelCol, HasPredictionCol):
def setParams(self, predictionCol="prediction", labelCol="label",
metricName="rmse"):
"""
- setParams(self, predictionCol="prediction", labelCol="label",
+ setParams(self, predictionCol="prediction", labelCol="label", \
metricName="rmse")
Sets params for regression evaluator.
"""