diff options
| author | DB Tsai <dbtsai@alpinenow.com> | 2014-12-29 17:17:12 -0800 |
|---|---|---|
| committer | Xiangrui Meng <meng@databricks.com> | 2014-12-29 17:17:12 -0800 |
| commit | 040d6f2d13b132b3ef2a1e4f12f9f0e781c5a0b8 (patch) | |
| tree | 0338b1a51e8139ec657383493a2b4ed753128e2b /mllib | |
| parent | 343db392b58fb33a3e4bc6fda1da69aaf686b5a9 (diff) | |
| download | spark-040d6f2d13b132b3ef2a1e4f12f9f0e781c5a0b8.tar.gz spark-040d6f2d13b132b3ef2a1e4f12f9f0e781c5a0b8.tar.bz2 spark-040d6f2d13b132b3ef2a1e4f12f9f0e781c5a0b8.zip | |
[SPARK-4972][MLlib] Updated the scala doc for lasso and ridge regression for the change of LeastSquaresGradient
In #SPARK-4907, we added factor of 2 into the LeastSquaresGradient. We updated the scala doc for lasso and ridge regression here.
Author: DB Tsai <dbtsai@alpinenow.com>
Closes #3808 from dbtsai/doc and squashes the following commits:
ec3c989 [DB Tsai] first commit
Diffstat (limited to 'mllib')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala | 2 | ||||
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala b/mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala index f9791c6571..8ecd5c6ad9 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/regression/Lasso.scala @@ -45,7 +45,7 @@ class LassoModel ( /** * Train a regression model with L1-regularization using Stochastic Gradient Descent. * This solves the l1-regularized least squares regression formulation - * f(weights) = 1/n ||A weights-y||^2 + regParam ||weights||_1 + * f(weights) = 1/2n ||A weights-y||^2 + regParam ||weights||_1 * Here the data matrix has n rows, and the input RDD holds the set of rows of A, each with * its corresponding right hand side label y. * See also the documentation for the precise formulation. diff --git a/mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala b/mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala index c8cad773f5..076ba35051 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/regression/RidgeRegression.scala @@ -45,7 +45,7 @@ class RidgeRegressionModel ( /** * Train a regression model with L2-regularization using Stochastic Gradient Descent. * This solves the l1-regularized least squares regression formulation - * f(weights) = 1/n ||A weights-y||^2 + regParam/2 ||weights||^2 + * f(weights) = 1/2n ||A weights-y||^2 + regParam/2 ||weights||^2 * Here the data matrix has n rows, and the input RDD holds the set of rows of A, each with * its corresponding right hand side label y. * See also the documentation for the precise formulation. |