Skip to content

Commit 19c2e85

Browse files
author
DB Tsai
committed
make stepsize twice to converge to the same solution
1 parent 0b2c29c commit 19c2e85

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mllib/src/test/scala/org/apache/spark/mllib/regression/StreamingLinearRegressionSuite.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ class StreamingLinearRegressionSuite extends FunSuite with TestSuiteBase {
5252
// create model
5353
val model = new StreamingLinearRegressionWithSGD()
5454
.setInitialWeights(Vectors.dense(0.0, 0.0))
55-
.setStepSize(0.1)
55+
.setStepSize(0.2)
5656
.setNumIterations(25)
5757

5858
// generate sequence of simulated data
@@ -84,7 +84,7 @@ class StreamingLinearRegressionSuite extends FunSuite with TestSuiteBase {
8484
// create model
8585
val model = new StreamingLinearRegressionWithSGD()
8686
.setInitialWeights(Vectors.dense(0.0))
87-
.setStepSize(0.1)
87+
.setStepSize(0.2)
8888
.setNumIterations(25)
8989

9090
// generate sequence of simulated data
@@ -118,7 +118,7 @@ class StreamingLinearRegressionSuite extends FunSuite with TestSuiteBase {
118118
// create model initialized with true weights
119119
val model = new StreamingLinearRegressionWithSGD()
120120
.setInitialWeights(Vectors.dense(10.0, 10.0))
121-
.setStepSize(0.1)
121+
.setStepSize(0.2)
122122
.setNumIterations(25)
123123

124124
// generate sequence of simulated data for testing

0 commit comments

Comments
 (0)