[SPARK-6705][MLLIB] Add fitInterceptTerm to LogisticRegression

Whoops, add this to the logisticRegression and not the optimizer
apache · Apr 4, 2015 · 9f1286b · 9f1286b
1 parent 1d6bd6f
commit 9f1286b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala
@@ -70,11 +70,11 @@ class LogisticRegression
     }
 
     // Train model
-    val lr = new LogisticRegressionWithLBFGS
+    val lr = new LogisticRegressionWithLBFGS()
+      .setIntercept(paramMap(fitIntercept))
     lr.optimizer
       .setRegParam(paramMap(regParam))
       .setNumIterations(paramMap(maxIter))
-      .addIntercept(paramMap(fitIntercept))
     val oldModel = lr.run(oldDataset)
     val lrm = new LogisticRegressionModel(this, paramMap, oldModel.weights, oldModel.intercept)