Repository: spark Updated Branches: refs/heads/branch-1.5 f4e125acf -> cff0fe291
[SPARK-9586] [ML] Update BinaryClassificationEvaluator to use setRawPredictionCol Update BinaryClassificationEvaluator to use setRawPredictionCol, rather than setScoreCol. Deprecated setScoreCol. I don't think setScoreCol was actually used anywhere (based on search). CC: mengxr Author: Joseph K. Bradley <jos...@databricks.com> Closes #7921 from jkbradley/binary-eval-rawpred and squashes the following commits: e5d7dfa [Joseph K. Bradley] Update BinaryClassificationEvaluator to use setRawPredictionCol (cherry picked from commit b77d3b9688d56d33737909375d1d0db07da5827b) Signed-off-by: Xiangrui Meng <m...@databricks.com> Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/cff0fe29 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/cff0fe29 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/cff0fe29 Branch: refs/heads/branch-1.5 Commit: cff0fe291aa470ef5cf4e5087c7114fb6360572f Parents: f4e125a Author: Joseph K. Bradley <jos...@databricks.com> Authored: Tue Aug 4 16:52:43 2015 -0700 Committer: Xiangrui Meng <m...@databricks.com> Committed: Tue Aug 4 16:52:53 2015 -0700 ---------------------------------------------------------------------- .../spark/ml/evaluation/BinaryClassificationEvaluator.scala | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/cff0fe29/mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala ---------------------------------------------------------------------- diff --git a/mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala b/mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala index 4a82b77..5d5cb7e 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala @@ -28,7 +28,7 @@ import org.apache.spark.sql.types.DoubleType /** * :: Experimental :: - * Evaluator for binary classification, which expects two input columns: score and label. + * Evaluator for binary classification, which expects two input columns: rawPrediction and label. */ @Experimental class BinaryClassificationEvaluator(override val uid: String) @@ -50,6 +50,13 @@ class BinaryClassificationEvaluator(override val uid: String) def setMetricName(value: String): this.type = set(metricName, value) /** @group setParam */ + def setRawPredictionCol(value: String): this.type = set(rawPredictionCol, value) + + /** + * @group setParam + * @deprecated use [[setRawPredictionCol()]] instead + */ + @deprecated("use setRawPredictionCol instead", "1.5.0") def setScoreCol(value: String): this.type = set(rawPredictionCol, value) /** @group setParam */ --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org For additional commands, e-mail: commits-h...@spark.apache.org