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 63
63
<dependency >
64
64
<groupId >org.scalanlp</groupId >
65
65
<artifactId >breeze_${scala.binary.version}</artifactId >
66
- <version >0.11</version >
66
+ <version >0.11.1 </version >
67
67
<exclusions >
68
68
<!-- This is included as a compile-scoped dependency by jtransforms, which is
69
69
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
372
372
testRDD2.cache()
373
373
testRDD3.cache()
374
374
375
+ val numIteration = 10
376
+
375
377
val lrA = new LogisticRegressionWithLBFGS ().setIntercept(true )
378
+ lrA.optimizer.setNumIterations(numIteration)
376
379
val lrB = new LogisticRegressionWithLBFGS ().setIntercept(true ).setFeatureScaling(false )
380
+ lrB.optimizer.setNumIterations(numIteration)
377
381
378
382
val modelA1 = lrA.run(testRDD1, initialWeights)
379
383
val modelA2 = lrA.run(testRDD2, initialWeights)
You can’t perform that action at this time.
0 commit comments