We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 397a208 commit 35c2f26Copy full SHA for 35c2f26
mllib/src/test/scala/org/apache/spark/mllib/classification/LogisticRegressionSuite.scala
@@ -372,8 +372,12 @@ class LogisticRegressionSuite extends FunSuite with MLlibTestSparkContext with M
372
testRDD2.cache()
373
testRDD3.cache()
374
375
+ val numIteration = 10
376
+
377
val lrA = new LogisticRegressionWithLBFGS().setIntercept(true)
378
+ lrA.optimizer.setNumIterations(numIteration)
379
val lrB = new LogisticRegressionWithLBFGS().setIntercept(true).setFeatureScaling(false)
380
+ lrB.optimizer.setNumIterations(numIteration)
381
382
val modelA1 = lrA.run(testRDD1, initialWeights)
383
val modelA2 = lrA.run(testRDD2, initialWeights)
0 commit comments