Reindent estimateCost's declaration and make it return Seqs.

apache · Apr 29, 2014 · 2ab7a5d · 2ab7a5d
1 parent 8b21e6d
commit 2ab7a5d
Showing 1 changed file with 8 additions and 4 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
@@ -727,9 +727,12 @@ object ALS {
    * @param productPartitioner  partitioner for partitioning products
    */
   @DeveloperApi
-  def estimateCost(ratings: RDD[Rating], rank: Int, userPartitioner: Partitioner,
-      productPartitioner: Partitioner):
-      (Map[Int, IterationCost], Map[Int, IterationCost]) = {
+  def estimateCost(
+      ratings: RDD[Rating],
+      rank: Int,
+      userPartitioner: Partitioner,
+      productPartitioner: Partitioner
+    ): (Seq[IterationCost], Seq[IterationCost]) = {
     // user partition -> set of products
     val utalk = ratings.mapPartitions(x => {
         val ht = new mutable.HashSet[(Int, Int)]()
@@ -823,7 +826,8 @@ object ALS {
     // We do two rank*rank outer products per rating and one Cholesky factorisation per user and
     // per product.
 
-    (userAnswer.toMap, productAnswer.toMap)
+    (   userAnswer.toArray.sortBy(x => x._1).map(x => x._2),
+     productAnswer.toArray.sortBy(x => x._1).map(x => x._2))
   }
 
   private class ALSRegistrator extends KryoRegistrator {