-
- Downloads
[SPARK-1327] GLM needs to check addIntercept for intercept and weights
GLM needs to check addIntercept for intercept and weights. The current implementation always uses the first weight as intercept. Added a test for training without adding intercept. JIRA: https://spark-project.atlassian.net/browse/SPARK-1327 Author: Xiangrui Meng <meng@databricks.com> Closes #236 from mengxr/glm and squashes the following commits: bcac1ac [Xiangrui Meng] add two tests to ensure {Lasso, Ridge}.setIntercept will throw an exceptions a104072 [Xiangrui Meng] remove protected to be compatible with 0.9 0e57aa4 [Xiangrui Meng] update Lasso and RidgeRegression to parse the weights correctly from GLM mark createModel protected mark predictPoint protected d7f629f [Xiangrui Meng] fix a bug in GLM when intercept is not used
Showing
- mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala 12 additions, 9 deletions...e/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
- mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala 14 additions, 6 deletions.../main/scala/org/apache/spark/mllib/regression/Lasso.scala
- mllib/src/main/scala/org/apache/spark/mllib/regression/LinearRegression.scala 10 additions, 10 deletions.../org/apache/spark/mllib/regression/LinearRegression.scala
- mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala 13 additions, 5 deletions...a/org/apache/spark/mllib/regression/RidgeRegression.scala
- mllib/src/test/scala/org/apache/spark/mllib/regression/LassoSuite.scala 6 additions, 3 deletions.../scala/org/apache/spark/mllib/regression/LassoSuite.scala
- mllib/src/test/scala/org/apache/spark/mllib/regression/LinearRegressionSuite.scala 25 additions, 1 deletion...apache/spark/mllib/regression/LinearRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/mllib/regression/RidgeRegressionSuite.scala 6 additions, 3 deletions.../apache/spark/mllib/regression/RidgeRegressionSuite.scala
Loading
Please register or sign in to comment