diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2016-02-21 20:20:41 -0800 |
---|---|---|
committer | DB Tsai <dbt@netflix.com> | 2016-02-21 20:20:41 -0800 |
commit | 8a4ed78869e99c7de7062c3baa0ddb9d28c8e9b1 (patch) | |
tree | 366cc15fbffd65feb502bc3c9de946f443df2165 /mllib/src/main | |
parent | 9bf6a926a1071fa59f76ab12b23df8de618f1cae (diff) | |
download | spark-8a4ed78869e99c7de7062c3baa0ddb9d28c8e9b1.tar.gz spark-8a4ed78869e99c7de7062c3baa0ddb9d28c8e9b1.tar.bz2 spark-8a4ed78869e99c7de7062c3baa0ddb9d28c8e9b1.zip |
[SPARK-13379][MLLIB] Fix MLlib LogisticRegressionWithLBFGS set regularization incorrectly
## What changes were proposed in this pull request?
Fix MLlib LogisticRegressionWithLBFGS regularization map as:
```SquaredL2Updater``` -> ```elasticNetParam = 0.0```
```L1Updater``` -> ```elasticNetParam = 1.0```
cc dbtsai
## How was the this patch tested?
unit tests
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #11258 from yanboliang/spark-13379.
Diffstat (limited to 'mllib/src/main')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala index bf68e3edd7..c3882606d7 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala @@ -444,8 +444,8 @@ class LogisticRegressionWithLBFGS createModel(weights, mlLogisticRegresionModel.intercept) } optimizer.getUpdater() match { - case x: SquaredL2Updater => runWithMlLogisitcRegression(1.0) - case x: L1Updater => runWithMlLogisitcRegression(0.0) + case x: SquaredL2Updater => runWithMlLogisitcRegression(0.0) + case x: L1Updater => runWithMlLogisitcRegression(1.0) case _ => super.run(input, initialWeights) } } else { |