Skip to content

Commit 7495681

Browse files
committed
fix compile error
1 parent 142a857 commit 7495681

File tree

1 file changed

+1
-1
lines changed
  • mllib/src/main/scala/org/apache/spark/mllib/optimization

1 file changed

+1
-1
lines changed

mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ object LBFGS extends Logging {
200200
val n = weights.length
201201
val bcWeights = data.context.broadcast(weights)
202202

203-
val (gradientSum, lossSum) = localData.treeAggregate((BDV.zeros[Double](n), 0.0))(
203+
val (gradientSum, lossSum) = data.treeAggregate((BDV.zeros[Double](n), 0.0))(
204204
seqOp = (c, v) => (c, v) match { case ((grad, loss), (label, features)) =>
205205
val l = localGradient.compute(
206206
features, label, Vectors.fromBreeze(bcWeights.value), Vectors.fromBreeze(grad))

0 commit comments

Comments
 (0)