From 502ce526fc8ec84fd2c1f3b2b9a74b07e76c2d65 Mon Sep 17 00:00:00 2001 From: Reza Zadeh Date: Mon, 4 Aug 2014 15:02:36 -0700 Subject: [PATCH] new interface --- .../apache/spark/mllib/linalg/distributed/RowMatrix.scala | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala index be92cbd33eba8..8c0f1eedfabd4 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala @@ -414,7 +414,7 @@ class RowMatrix( require(colMags.size == this.numCols(), "Number of magnitudes didn't match column dimension") - val sg = math.sqrt(gamma) + val sg = math.sqrt(gamma) // sqrt(gamma) used many times val sims = rows.flatMap { row => @@ -427,7 +427,8 @@ class RowMatrix( case (_, 0.0) => // Skip explicit zero elements. case (j, jVal) => if (Math.random < sg / colMags(j)) { - val contrib = ((i.toLong, j.toLong), (iVal * jVal) / (math.min(sg, colMags(i)) * math.min(sg, colMags(j)))) + val contrib = ((i.toLong, j.toLong), (iVal * jVal) / + (math.min(sg, colMags(i)) * math.min(sg, colMags(j)))) buf += contrib } }