diff options
| -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))) }) }) |