diff options
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.scala | 7 |
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 } |