File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed
src/test/scala/org/apache/spark/mllib/classification Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change 6363 <dependency >
6464 <groupId >org.scalanlp</groupId >
6565 <artifactId >breeze_${scala.binary.version}</artifactId >
66- <version >0.11</version >
66+ <version >0.11.1 </version >
6767 <exclusions >
6868 <!-- This is included as a compile-scoped dependency by jtransforms, which is
6969 a dependency of breeze. -->
Original file line number Diff line number Diff 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)
You can’t perform that action at this time.
0 commit comments