-
- Downloads
Refactor optimizers and create GLMs
This change refactors the structure of GLMs to use mixins which maintain a similar interface to other ML lib algorithms. This change also creates an Optimizer trait which allows GLMs to be extended to use other optimization techniques.
Showing
- mllib/src/main/scala/spark/mllib/classification/LogisticRegression.scala 30 additions, 63 deletions...scala/spark/mllib/classification/LogisticRegression.scala
- mllib/src/main/scala/spark/mllib/classification/SVM.scala 26 additions, 62 deletionsmllib/src/main/scala/spark/mllib/classification/SVM.scala
- mllib/src/main/scala/spark/mllib/optimization/GradientDescent.scala 72 additions, 8 deletions...main/scala/spark/mllib/optimization/GradientDescent.scala
- mllib/src/main/scala/spark/mllib/optimization/Optimizer.scala 29 additions, 0 deletions...b/src/main/scala/spark/mllib/optimization/Optimizer.scala
- mllib/src/main/scala/spark/mllib/regression/GeneralizedLinearAlgorithm.scala 125 additions, 0 deletions...a/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
- mllib/src/main/scala/spark/mllib/regression/Lasso.scala 27 additions, 62 deletionsmllib/src/main/scala/spark/mllib/regression/Lasso.scala
- mllib/src/test/scala/spark/mllib/classification/LogisticRegressionSuite.scala 2 additions, 4 deletions.../spark/mllib/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/spark/mllib/classification/SVMSuite.scala 4 additions, 5 deletions.../src/test/scala/spark/mllib/classification/SVMSuite.scala
- mllib/src/test/scala/spark/mllib/regression/LassoSuite.scala 5 additions, 6 deletionsmllib/src/test/scala/spark/mllib/regression/LassoSuite.scala
Loading
Please register or sign in to comment