From 0b5c3e306971cc5ed27eaa8aa3b455585118ae63 Mon Sep 17 00:00:00 2001 From: GuoQiang Li Date: Fri, 21 Nov 2014 15:27:06 +0800 Subject: [PATCH] Minor fix --- .../org/apache/spark/mllib/optimization/GradientDescent.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala b/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala index 97d477bad7e71..9061c093f0bd3 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala @@ -187,7 +187,7 @@ object GradientDescent extends Logging { val (gradientSum, lossSum, miniBatchSize) = data.sample(false, miniBatchFraction, 42 + i) .treeAggregate((BDV.zeros[Double](n), 0.0, 0.0))( seqOp = (c, v) => { - // c: (grad, loss), v: (label, features) + // c: (grad, loss, count), v: (label, features) val l = gradient.compute(v._2, v._1, bcWeights.value, Vectors.fromBreeze(c._1)) (c._1, c._2 + l, c._3 + 1) },