aboutsummaryrefslogtreecommitdiff
path: root/graph/src
diff options
context:
space:
mode:
authorJianping J Wang <jianping.j.wang@gmail.com>2013-12-31 17:11:20 +0800
committerJianping J Wang <jianping.j.wang@gmail.com>2013-12-31 17:11:20 +0800
commitab7b8ce13ec3a0a85f57ceede02df1e5eb9f6c23 (patch)
tree65999a0b4acdd17c746fe2de3436efd50eb5edab /graph/src
parent4a30f69b25239d0e7a4a27478be261e17c1dba1a (diff)
downloadspark-ab7b8ce13ec3a0a85f57ceede02df1e5eb9f6c23.tar.gz
spark-ab7b8ce13ec3a0a85f57ceede02df1e5eb9f6c23.tar.bz2
spark-ab7b8ce13ec3a0a85f57ceede02df1e5eb9f6c23.zip
Update Svdpp.scala
Diffstat (limited to 'graph/src')
-rw-r--r--graph/src/main/scala/org/apache/spark/graph/algorithms/Svdpp.scala2
1 files changed, 1 insertions, 1 deletions
diff --git a/graph/src/main/scala/org/apache/spark/graph/algorithms/Svdpp.scala b/graph/src/main/scala/org/apache/spark/graph/algorithms/Svdpp.scala
index 7c3e0c83c9..cb1a69e318 100644
--- a/graph/src/main/scala/org/apache/spark/graph/algorithms/Svdpp.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/algorithms/Svdpp.scala
@@ -71,7 +71,7 @@ object Svdpp {
}
for (i <- 0 until conf.maxIters) {
- // phase 1, calculate v2 for user nodes
+ // phase 1, calculate pu + |N(u)|^(-0.5)*sum(y) for user nodes
var t1 = g.mapReduceTriplets(et => Iterator((et.srcId, et.dstAttr._2)), (g1: RealVector, g2: RealVector) => g1.add(g2))
g = g.outerJoinVertices(t1) { (vid: Vid, vd: (RealVector, RealVector, Double, Double), msg: Option[RealVector]) =>
if (msg.isDefined) (vd._1, vd._1.add(msg.get.mapMultiply(vd._4)), vd._3, vd._4) else vd