-
- Downloads
[SPARK-6791][ML] Add read/write for CrossValidator and Evaluators
I believe this works for general estimators within CrossValidator, including compound estimators. (See the complex unit test.) Added read/write for all 3 Evaluators as well. CC: mengxr yanboliang Author: Joseph K. Bradley <joseph@databricks.com> Closes #9848 from jkbradley/cv-io.
Showing
- mllib/src/main/scala/org/apache/spark/ml/Pipeline.scala 3 additions, 35 deletionsmllib/src/main/scala/org/apache/spark/ml/Pipeline.scala
- mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala 9 additions, 2 deletions...e/spark/ml/evaluation/BinaryClassificationEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluator.scala 10 additions, 2 deletions...ark/ml/evaluation/MulticlassClassificationEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/evaluation/RegressionEvaluator.scala 9 additions, 2 deletions.../org/apache/spark/ml/evaluation/RegressionEvaluator.scala
- mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala 3 additions, 11 deletions...c/main/scala/org/apache/spark/ml/recommendation/ALS.scala
- mllib/src/main/scala/org/apache/spark/ml/tuning/CrossValidator.scala 223 additions, 6 deletions...ain/scala/org/apache/spark/ml/tuning/CrossValidator.scala
- mllib/src/main/scala/org/apache/spark/ml/util/ReadWrite.scala 34 additions, 14 deletions...b/src/main/scala/org/apache/spark/ml/util/ReadWrite.scala
- mllib/src/test/scala/org/apache/spark/ml/PipelineSuite.scala 1 addition, 3 deletionsmllib/src/test/scala/org/apache/spark/ml/PipelineSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluatorSuite.scala 12 additions, 1 deletion...rk/ml/evaluation/BinaryClassificationEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluatorSuite.scala 12 additions, 1 deletion...l/evaluation/MulticlassClassificationEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala 11 additions, 1 deletion...apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/tuning/CrossValidatorSuite.scala 195 additions, 7 deletions...cala/org/apache/spark/ml/tuning/CrossValidatorSuite.scala
Loading
Please register or sign in to comment