SPARK-6710 GraphX Fixed Wrong initial bias in GraphX SVDPlusPlus

Author: Michael Malak <[email protected]> Closes apache#5464 from michaelmalak/master and squashes the following commits: 9d942ba [Michael Malak] SPARK-6710 GraphX Fixed Wrong initial bias in GraphX SVDPlusPlus
yunmingzhang17 · Apr 12, 2015 · 1205f7e · 1205f7e
1 parent dea5dac
commit 1205f7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/lib/SVDPlusPlus.scala b/graphx/src/main/scala/org/apache/spark/graphx/lib/SVDPlusPlus.scala
@@ -93,7 +93,7 @@ object SVDPlusPlus {
     val gJoinT0 = g.outerJoinVertices(t0) {
       (vid: VertexId, vd: (Array[Double], Array[Double], Double, Double),
        msg: Option[(Long, Double)]) =>
-        (vd._1, vd._2, msg.get._2 / msg.get._1, 1.0 / scala.math.sqrt(msg.get._1))
+        (vd._1, vd._2, msg.get._2 / msg.get._1 - u, 1.0 / scala.math.sqrt(msg.get._1))
     }.cache()
     materialize(gJoinT0)
     g.unpersist()