-
- Downloads
[SPARK-13962][ML] spark.ml Evaluators should support other numeric types for label
## What changes were proposed in this pull request? Made BinaryClassificationEvaluator, MulticlassClassificationEvaluator and RegressionEvaluator accept all numeric types for label ## How was this patch tested? Unit tests Author: BenFradet <benjamin.fradet@gmail.com> Closes #12500 from BenFradet/SPARK-13962.
Showing
- mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala 7 additions, 5 deletions...e/spark/ml/evaluation/BinaryClassificationEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluator.scala 6 additions, 5 deletions...ark/ml/evaluation/MulticlassClassificationEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/evaluation/RegressionEvaluator.scala 5 additions, 14 deletions.../org/apache/spark/ml/evaluation/RegressionEvaluator.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluatorSuite.scala 6 additions, 1 deletion...rk/ml/evaluation/BinaryClassificationEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluatorSuite.scala 5 additions, 1 deletion...l/evaluation/MulticlassClassificationEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala 5 additions, 1 deletion...apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/tree/impl/TreeTests.scala 7 additions, 2 deletions.../test/scala/org/apache/spark/ml/tree/impl/TreeTests.scala
- mllib/src/test/scala/org/apache/spark/ml/util/MLTestingUtils.scala 47 additions, 22 deletions.../test/scala/org/apache/spark/ml/util/MLTestingUtils.scala
Loading
Please register or sign in to comment