diff options
author | baishuo(白硕) <vc_java@hotmail.com> | 2014-05-23 13:02:40 -0700 |
---|---|---|
committer | Reynold Xin <rxin@apache.org> | 2014-05-23 13:02:40 -0700 |
commit | a08262d8769808dd3a8ee1b1e80fbf6ac13a557c (patch) | |
tree | e4026cf6fee57d5b6c21bacc61c4bccca14d274a /mllib | |
parent | b2bdd0e505f1ae3d39c46139f17bd43779ece635 (diff) | |
download | spark-a08262d8769808dd3a8ee1b1e80fbf6ac13a557c.tar.gz spark-a08262d8769808dd3a8ee1b1e80fbf6ac13a557c.tar.bz2 spark-a08262d8769808dd3a8ee1b1e80fbf6ac13a557c.zip |
Update LBFGSSuite.scala
the same reason as https://github.com/apache/spark/pull/588
Author: baishuo(白硕) <vc_java@hotmail.com>
Closes #815 from baishuo/master and squashes the following commits:
6876c1e [baishuo(白硕)] Update LBFGSSuite.scala
Diffstat (limited to 'mllib')
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/mllib/optimization/LBFGSSuite.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/optimization/LBFGSSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/optimization/LBFGSSuite.scala index 6af1b502eb..820eca9b1b 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/optimization/LBFGSSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/optimization/LBFGSSuite.scala @@ -43,7 +43,7 @@ class LBFGSSuite extends FunSuite with LocalSparkContext with ShouldMatchers { // Add an extra variable consisting of all 1.0's for the intercept. val testData = GradientDescentSuite.generateGDInput(A, B, nPoints, 42) val data = testData.map { case LabeledPoint(label, features) => - label -> Vectors.dense(1.0, features.toArray: _*) + label -> Vectors.dense(1.0 +: features.toArray) } lazy val dataRDD = sc.parallelize(data, 2).cache() @@ -55,7 +55,7 @@ class LBFGSSuite extends FunSuite with LocalSparkContext with ShouldMatchers { test("LBFGS loss should be decreasing and match the result of Gradient Descent.") { val regParam = 0 - val initialWeightsWithIntercept = Vectors.dense(1.0, initialWeights: _*) + val initialWeightsWithIntercept = Vectors.dense(1.0 +: initialWeights.toArray) val convergenceTol = 1e-12 val maxNumIterations = 10 |