aboutsummaryrefslogtreecommitdiff
path: root/mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala
diff options
context:
space:
mode:
Diffstat (limited to 'mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala')
-rw-r--r--mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala7
1 files changed, 6 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala b/mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala
index 13bd3307f8..0f22cca3a7 100644
--- a/mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala
+++ b/mllib/src/main/scala/org/apache/spark/ml/evaluation/Evaluator.scala
@@ -17,7 +17,7 @@
package org.apache.spark.ml.evaluation
-import org.apache.spark.annotation.DeveloperApi
+import org.apache.spark.annotation.{DeveloperApi, Since}
import org.apache.spark.ml.param.{ParamMap, Params}
import org.apache.spark.sql.DataFrame
@@ -25,6 +25,7 @@ import org.apache.spark.sql.DataFrame
* :: DeveloperApi ::
* Abstract class for evaluators that compute metrics from predictions.
*/
+@Since("1.5.0")
@DeveloperApi
abstract class Evaluator extends Params {
@@ -35,6 +36,7 @@ abstract class Evaluator extends Params {
* @param paramMap parameter map that specifies the input columns and output metrics
* @return metric
*/
+ @Since("1.5.0")
def evaluate(dataset: DataFrame, paramMap: ParamMap): Double = {
this.copy(paramMap).evaluate(dataset)
}
@@ -44,6 +46,7 @@ abstract class Evaluator extends Params {
* @param dataset a dataset that contains labels/observations and predictions.
* @return metric
*/
+ @Since("1.5.0")
def evaluate(dataset: DataFrame): Double
/**
@@ -51,7 +54,9 @@ abstract class Evaluator extends Params {
* or minimized (false).
* A given evaluator may support multiple metrics which may be maximized or minimized.
*/
+ @Since("1.5.0")
def isLargerBetter: Boolean = true
+ @Since("1.5.0")
override def copy(extra: ParamMap): Evaluator
}