For comments.

apache · Feb 11, 2015 · 4870fe4 · 4870fe4
1 parent 9af1487
commit 4870fe4
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDA.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/LDA.scala
@@ -456,13 +456,11 @@ private[clustering] object LDA {
           val random = new Random(partIndex + randomSeed)
           partEdges.flatMap { edge =>
             val gamma = normalize(BDV.fill[Double](k)(random.nextDouble()), 1.0)
-            val sum = BDV.zeros[Double](k)
-            brzAxpy(edge.attr, gamma, sum)
-
+            val sum = gamma * edge.attr
             Seq((edge.srcId, sum), (edge.dstId, sum))
           }
         }
-      verticesTMP.reduceByKey((sum0, sum1) => { sum0 + sum1 })
+      verticesTMP.reduceByKey(_ + _)
     }
 
     val docTermVertices = createVertices()