Skip to content

Commit 35c2f26

Browse files
committed
fix LRSuite
1 parent 397a208 commit 35c2f26

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

mllib/src/test/scala/org/apache/spark/mllib/classification/LogisticRegressionSuite.scala

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -372,8 +372,12 @@ class LogisticRegressionSuite extends FunSuite with MLlibTestSparkContext with M
372372
testRDD2.cache()
373373
testRDD3.cache()
374374

375+
val numIteration = 10
376+
375377
val lrA = new LogisticRegressionWithLBFGS().setIntercept(true)
378+
lrA.optimizer.setNumIterations(numIteration)
376379
val lrB = new LogisticRegressionWithLBFGS().setIntercept(true).setFeatureScaling(false)
380+
lrB.optimizer.setNumIterations(numIteration)
377381

378382
val modelA1 = lrA.run(testRDD1, initialWeights)
379383
val modelA2 = lrA.run(testRDD2, initialWeights)

0 commit comments

Comments
 (0)