Skip to content

Commit 9b30c5c

Browse files
author
Anthony Truchet
committed
Fix error triggered in PySpark
1 parent 0ce8c64 commit 9b30c5c

File tree

1 file changed

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

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,8 +254,10 @@ object LBFGS extends Logging {
254254
// Adds two (gradient, loss) tuples
255255
val combOp = (c1: (Vector, Double), c2: (Vector, Double)) =>
256256
(c1, c2) match { case ((grad1, loss1), (grad2, loss2)) =>
257-
axpy(1.0, grad2, grad1)
258-
(grad1, loss1 + loss2)
257+
val denseGrad1 = grad1.toDense
258+
val denseGrad2 = grad2.toDense
259+
axpy(1.0, denseGrad2, denseGrad1)
260+
(denseGrad1, loss1 + loss2)
259261
}
260262

261263
val zeroSparseVector = Vectors.sparse(n, Seq())

0 commit comments

Comments
 (0)