diff options
author | Nick Pentreath <nick.pentreath@gmail.com> | 2013-10-04 13:52:53 +0200 |
---|---|---|
committer | Nick Pentreath <nick.pentreath@gmail.com> | 2013-10-04 13:52:53 +0200 |
commit | c6ceaeae50c7b5d69bb9897af32f537bfbde152d (patch) | |
tree | 597e13ba04ca6a493c10a29d37f504e5748f8a2d | |
parent | 6a7836cddcf3ae0322e76cd5f79f6dd9ea73a09c (diff) | |
download | spark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.tar.gz spark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.tar.bz2 spark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.zip |
Style fix using 'if' rather than 'match' on boolean
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala | 27 |
1 files changed, 13 insertions, 14 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala index 5935d2754d..36853acab5 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala @@ -207,20 +207,19 @@ class ALS private (var numBlocks: Int, var rank: Int, var iterations: Int, var l * @return Option[YtY] - whose value is only used in the implicit preference model */ def computeYtY(factors: RDD[(Int, Array[Array[Double]])]) = { - implicitPrefs match { - case true => { - Option( - factors.flatMapValues{ case factorArray => - factorArray.map{ vector => - val x = new DoubleMatrix(vector) - x.mmul(x.transpose()) - } - }.reduceByKeyLocally((a, b) => a.addi(b)) - .values - .reduce((a, b) => a.addi(b)) - ) - } - case false => None + if (implicitPrefs) { + Option( + factors.flatMapValues{ case factorArray => + factorArray.map{ vector => + val x = new DoubleMatrix(vector) + x.mmul(x.transpose()) + } + }.reduceByKeyLocally((a, b) => a.addi(b)) + .values + .reduce((a, b) => a.addi(b)) + ) + } else { + None } } |