-
- Downloads
Refactor GLM algorithms and add Java tests
This change adds Java examples and unit tests for all GLM algorithms to make sure the MLLib interface works from Java. Changes include - Introduce LabeledPoint and avoid using Doubles in train arguments - Rename train to run in class methods - Make the optimizer a member variable of GLM to make sure the builder pattern works
Showing
- examples/src/main/java/spark/mllib/examples/JavaLR.java 85 additions, 0 deletionsexamples/src/main/java/spark/mllib/examples/JavaLR.java
- mllib/src/main/scala/spark/mllib/classification/ClassificationModel.scala 2 additions, 2 deletions...cala/spark/mllib/classification/ClassificationModel.scala
- mllib/src/main/scala/spark/mllib/classification/LogisticRegression.scala 18 additions, 16 deletions...scala/spark/mllib/classification/LogisticRegression.scala
- mllib/src/main/scala/spark/mllib/classification/SVM.scala 14 additions, 11 deletionsmllib/src/main/scala/spark/mllib/classification/SVM.scala
- mllib/src/main/scala/spark/mllib/optimization/GradientDescent.scala 5 additions, 8 deletions...main/scala/spark/mllib/optimization/GradientDescent.scala
- mllib/src/main/scala/spark/mllib/regression/GeneralizedLinearAlgorithm.scala 26 additions, 34 deletions...a/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
- mllib/src/main/scala/spark/mllib/regression/LabeledPoint.scala 32 additions, 0 deletions.../src/main/scala/spark/mllib/regression/LabeledPoint.scala
- mllib/src/main/scala/spark/mllib/regression/Lasso.scala 13 additions, 9 deletionsmllib/src/main/scala/spark/mllib/regression/Lasso.scala
- mllib/src/main/scala/spark/mllib/regression/RidgeRegression.scala 4 additions, 3 deletions...c/main/scala/spark/mllib/regression/RidgeRegression.scala
- mllib/src/main/scala/spark/mllib/util/LassoDataGenerator.scala 2 additions, 2 deletions.../src/main/scala/spark/mllib/util/LassoDataGenerator.scala
- mllib/src/main/scala/spark/mllib/util/LogisticRegressionDataGenerator.scala 3 additions, 2 deletions...la/spark/mllib/util/LogisticRegressionDataGenerator.scala
- mllib/src/main/scala/spark/mllib/util/MLUtils.scala 5 additions, 4 deletionsmllib/src/main/scala/spark/mllib/util/MLUtils.scala
- mllib/src/main/scala/spark/mllib/util/RidgeRegressionDataGenerator.scala 4 additions, 3 deletions...scala/spark/mllib/util/RidgeRegressionDataGenerator.scala
- mllib/src/main/scala/spark/mllib/util/SVMDataGenerator.scala 3 additions, 2 deletionsmllib/src/main/scala/spark/mllib/util/SVMDataGenerator.scala
- mllib/src/test/scala/spark/mllib/classification/JavaLogisticRegressionSuite.java 98 additions, 0 deletions...ark/mllib/classification/JavaLogisticRegressionSuite.java
- mllib/src/test/scala/spark/mllib/classification/JavaSVMSuite.java 98 additions, 0 deletions...c/test/scala/spark/mllib/classification/JavaSVMSuite.java
- mllib/src/test/scala/spark/mllib/classification/LogisticRegressionSuite.scala 38 additions, 24 deletions.../spark/mllib/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/spark/mllib/classification/SVMSuite.scala 39 additions, 23 deletions.../src/test/scala/spark/mllib/classification/SVMSuite.scala
- mllib/src/test/scala/spark/mllib/regression/JavaLassoSuite.java 96 additions, 0 deletions...src/test/scala/spark/mllib/regression/JavaLassoSuite.java
- mllib/src/test/scala/spark/mllib/regression/LassoSuite.scala 39 additions, 25 deletionsmllib/src/test/scala/spark/mllib/regression/LassoSuite.scala
Loading
Please register or sign in to comment