-
- Downloads
[SPARK-11763][ML] Add save,load to LogisticRegression Estimator
Add save/load to LogisticRegression Estimator, and refactor tests a little to make it easier to add similar support to other Estimator, Model pairs. Moved LogisticRegressionReader/Writer to within LogisticRegressionModel CC: mengxr Author: Joseph K. Bradley <joseph@databricks.com> Closes #9749 from jkbradley/lr-io-2.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 49 additions, 42 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/util/ReadWrite.scala 1 addition, 0 deletions...b/src/main/scala/org/apache/spark/ml/util/ReadWrite.scala
- mllib/src/test/scala/org/apache/spark/ml/PipelineSuite.scala 0 additions, 7 deletionsmllib/src/test/scala/org/apache/spark/ml/PipelineSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/ClassifierSuite.scala 32 additions, 0 deletions.../org/apache/spark/ml/classification/ClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala 28 additions, 9 deletions...che/spark/ml/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/ProbabilisticClassifierSuite.scala 14 additions, 0 deletions...park/ml/classification/ProbabilisticClassifierSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala 49 additions, 1 deletion...scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala
Loading
Please register or sign in to comment