Skip to content

Commit

Permalink
[SPARK-10336][example] fix not being able to set intercept in LR example
Browse files Browse the repository at this point in the history
`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

(cherry picked from commit 4572321)
Signed-off-by: DB Tsai <dbt@netflix.com>
  • Loading branch information
Shuo Xiang authored and DB Tsai committed Aug 28, 2015
1 parent ccda27a commit 9c58f64
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 9c58f64

Please sign in to comment.