-
- Downloads
Refactor SGD options into a new class.
This refactoring pulls out code shared between SVM, Lasso, LR into a common GradientDescentOpts class. Some style cleanup as well
Showing
- mllib/src/main/scala/spark/mllib/GradientDescentOpts.scala 76 additions, 0 deletionsmllib/src/main/scala/spark/mllib/GradientDescentOpts.scala
- mllib/src/main/scala/spark/mllib/classification/LogisticRegression.scala 10 additions, 38 deletions...scala/spark/mllib/classification/LogisticRegression.scala
- mllib/src/main/scala/spark/mllib/classification/SVM.scala 10 additions, 48 deletionsmllib/src/main/scala/spark/mllib/classification/SVM.scala
- mllib/src/main/scala/spark/mllib/optimization/GradientDescent.scala 8 additions, 11 deletions...main/scala/spark/mllib/optimization/GradientDescent.scala
- mllib/src/main/scala/spark/mllib/regression/Lasso.scala 10 additions, 48 deletionsmllib/src/main/scala/spark/mllib/regression/Lasso.scala
- mllib/src/test/scala/spark/mllib/classification/LogisticRegressionSuite.scala 5 additions, 2 deletions.../spark/mllib/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/spark/mllib/classification/SVMSuite.scala 15 additions, 8 deletions.../src/test/scala/spark/mllib/classification/SVMSuite.scala
- mllib/src/test/scala/spark/mllib/regression/LassoSuite.scala 9 additions, 4 deletionsmllib/src/test/scala/spark/mllib/regression/LassoSuite.scala
Loading
Please register or sign in to comment