diff options
-rw-r--r-- | mllib/src/main/scala/spark/mllib/recommendation/ALS.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala b/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala index aeacbca510..9097f46db9 100644 --- a/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala +++ b/mllib/src/main/scala/spark/mllib/recommendation/ALS.scala @@ -133,9 +133,9 @@ class ALS private (var numBlocks: Int, var rank: Int, var iterations: Int, var l var products = productOutLinks.mapPartitions {itr => val rand = new Random() itr.map({case (x, y) => - (x, y.elementIds.map(u => randomFactor(rank, rand))) + (x, y.elementIds.map(u => randomFactor(rank, rand))) }) - }) + } for (iter <- 0 until iterations) { // perform ALS update |