-
- Downloads
[SPARK-9716][ML] BinaryClassificationEvaluator should accept Double prediction column
This PR aims to allow the prediction column of `BinaryClassificationEvaluator` to be of double type. Author: BenFradet <benjamin.fradet@gmail.com> Closes #10472 from BenFradet/SPARK-9716.
Showing
- mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala 6 additions, 3 deletions...e/spark/ml/evaluation/BinaryClassificationEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/util/SchemaUtils.scala 17 additions, 0 deletions...src/main/scala/org/apache/spark/ml/util/SchemaUtils.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluatorSuite.scala 32 additions, 0 deletions...rk/ml/evaluation/BinaryClassificationEvaluatorSuite.scala
- python/pyspark/ml/evaluation.py 3 additions, 2 deletionspython/pyspark/ml/evaluation.py
Loading
Please register or sign in to comment