diff options
author | Holden Karau <holden@pigscanfly.ca> | 2013-08-12 22:13:08 -0700 |
---|---|---|
committer | Holden Karau <holden@pigscanfly.ca> | 2013-08-12 22:13:08 -0700 |
commit | d145da818ee14c01ff08b29a546fb3c87016aa51 (patch) | |
tree | 4e7595f8be88a94855806f67c8aa43c84ab33440 /mllib | |
parent | 705c9ace2a893168aadfca7d80749f3597d9a24a (diff) | |
download | spark-d145da818ee14c01ff08b29a546fb3c87016aa51.tar.gz spark-d145da818ee14c01ff08b29a546fb3c87016aa51.tar.bz2 spark-d145da818ee14c01ff08b29a546fb3c87016aa51.zip |
Code review feedback :)
Diffstat (limited to 'mllib')
-rw-r--r-- | mllib/src/main/scala/spark/mllib/recommendation/ALS.scala | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala b/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala index 974046d260..aeacbca510 100644 --- a/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala +++ b/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala @@ -124,16 +124,16 @@ class ALS private (var numBlocks: Int, var rank: Int, var iterations: Int, var l val (productInLinks, productOutLinks) = makeLinkRDDs(numBlocks, ratingsByProductBlock) // Initialize user and product factors randomly - var users = userOutLinks.mapPartitions(itr => { + var users = userOutLinks.mapPartitions {itr => val rand = new Random() - itr.map({case (x,y) => - (x,y.elementIds.map(u => randomFactor(rank, rand))) + itr.map({case (x, y) => + (x, y.elementIds.map(u => randomFactor(rank, rand))) }) - }) - var products = productOutLinks.mapPartitions(itr => { + } + var products = productOutLinks.mapPartitions {itr => val rand = new Random() - itr.map({case (x,y) => - (x,y.elementIds.map(u => randomFactor(rank, rand))) + itr.map({case (x, y) => + (x, y.elementIds.map(u => randomFactor(rank, rand))) }) }) |