diff options
| author | Shuo Xiang <sxiang@pinterest.com> | 2015-08-28 13:09:13 -0700 |
|---|---|---|
| committer | DB Tsai <dbt@netflix.com> | 2015-08-28 13:09:13 -0700 |
| commit | 45723214e694b9a440723e9504c562e6393709f3 (patch) | |
| tree | 6208602625c418ae54a33e767cd0989cdb1fcf4e /examples | |
| parent | c53c902fa9c458200245f919067b41dde9cd9418 (diff) | |
| download | spark-45723214e694b9a440723e9504c562e6393709f3.tar.gz spark-45723214e694b9a440723e9504c562e6393709f3.tar.bz2 spark-45723214e694b9a440723e9504c562e6393709f3.zip | |
[SPARK-10336][example] fix not being able to set intercept in LR example
`fitIntercept` is a command line option but not set in the main program.
dbtsai
Author: Shuo Xiang <sxiang@pinterest.com>
Closes #8510 from coderxiang/intercept and squashes the following commits:
57c9b7d [Shuo Xiang] fix not being able to set intercept in LR example
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/src/main/scala/org/apache/spark/examples/ml/LogisticRegressionExample.scala | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/examples/src/main/scala/org/apache/spark/examples/ml/LogisticRegressionExample.scala b/examples/src/main/scala/org/apache/spark/examples/ml/LogisticRegressionExample.scala index 7682557127..8e3760ddb5 100644 --- a/examples/src/main/scala/org/apache/spark/examples/ml/LogisticRegressionExample.scala +++ b/examples/src/main/scala/org/apache/spark/examples/ml/LogisticRegressionExample.scala @@ -136,6 +136,7 @@ object LogisticRegressionExample { .setElasticNetParam(params.elasticNetParam) .setMaxIter(params.maxIter) .setTol(params.tol) + .setFitIntercept(params.fitIntercept) stages += lor val pipeline = new Pipeline().setStages(stages.toArray) |