aboutsummaryrefslogtreecommitdiff
path: root/mllib
diff options
context:
space:
mode:
authorNick Pentreath <nick.pentreath@gmail.com>2013-10-04 13:52:53 +0200
committerNick Pentreath <nick.pentreath@gmail.com>2013-10-04 13:52:53 +0200
commitc6ceaeae50c7b5d69bb9897af32f537bfbde152d (patch)
tree597e13ba04ca6a493c10a29d37f504e5748f8a2d /mllib
parent6a7836cddcf3ae0322e76cd5f79f6dd9ea73a09c (diff)
downloadspark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.tar.gz
spark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.tar.bz2
spark-c6ceaeae50c7b5d69bb9897af32f537bfbde152d.zip
Style fix using 'if' rather than 'match' on boolean
Diffstat (limited to 'mllib')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala27
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
}
}